fix(test): set high LOS for stroke mortality in test_long_los, and add 1_in_n_delay to expected output in test_get_occupancy_freq (both of these are related to recent changes in the model, which altered the LOS param and output table)

Author: amyheather

tests/test_functionaltest.py

@@ -98,6 +98,7 @@ def test_long_los(stroke_no_esd_mean):
     param = Param(
         asu_los=ASULOS(stroke_no_esd_mean=stroke_no_esd_mean,
                        stroke_esd_mean=los,
+                       stroke_mortality_mean=los,
                        tia_mean=los,
                        neuro_mean=los,
                        other_mean=los),

tests/test_unittest.py

@@ -231,6 +231,7 @@ def test_get_occupancy_freq():
     expected_pct = [0.4, 0.3, 0.2, 0.1]
     expected_c_pct = [0.4, 0.7, 0.9, 1.0]
     expected_prob_delay = [1.0, 0.3/0.7, 0.2/0.9, 0.1/1.0]
+    expected_1_in_n_delay = [1, 2, 4, 10]
 
     # Create a Runner instance
     runner = Runner(None)
@@ -240,16 +241,17 @@ def test_get_occupancy_freq():
 
     # Check the structure of the DataFrame
     assert list(result_df.columns) == [
-        "beds", "freq", "pct", "c_pct", "prob_delay"]
+        "beds", "freq", "pct", "c_pct", "prob_delay", "1_in_n_delay"]
 
     # Check the values
     assert list(result_df["beds"]) == expected_beds
     assert list(result_df["freq"]) == expected_freq
     assert np.allclose(result_df["pct"], expected_pct)
     assert np.allclose(result_df["c_pct"], expected_c_pct)
 
-    # Check prob_delay calculation
+    # Check prob_delay and 1 in n delay calculations
     assert np.allclose(result_df["prob_delay"], expected_prob_delay)
+    assert np.allclose(result_df["1_in_n_delay"], expected_1_in_n_delay)
 
 
 # -----------------------------------------------------------------------------