test: Use parameter values rather than fixed constants to generalize test to all cases

PhilMiller · PhilMiller · commit b8f5e458dd2e · 2020-04-07T20:57:15.000-04:00
diff --git a/tests/penn_chime/test_models.py b/tests/penn_chime/test_models.py
@@ -196,6 +196,6 @@ def test_model_cumulative_census(param, model):
 
     # TODO: is 1.0 for ceil function?
     diff = admits.hospitalized[1:-1] - (
-        0.05 * 0.05 * (raw_df.infected[1:-1] + raw_df.recovered[1:-1]) - 1.0
+        param.market_share * param.hospitalized.rate * (raw_df.infected[1:-1] + raw_df.recovered[1:-1]) - 1.0
     )
     assert (diff.abs() < 0.1).all()

Original file line number	Diff line number	Diff line change
`@@ -196,6 +196,6 @@ def test_model_cumulative_census(param, model):`
`196`	`196`
`197`	`197`	`# TODO: is 1.0 for ceil function?`
`198`	`198`	`diff = admits.hospitalized[1:-1] - (`
`199`		`- 0.05 * 0.05 * (raw_df.infected[1:-1] + raw_df.recovered[1:-1]) - 1.0`
	`199`	`+ param.market_share * param.hospitalized.rate * (raw_df.infected[1:-1] + raw_df.recovered[1:-1]) - 1.0`
`200`	`200`	`)`
`201`	`201`	`assert (diff.abs() < 0.1).all()`