Skip to content

Commit d247c6e

Browse files
committed
Set mitigation application date explicitly in tests, so their results won't change when the model does
1 parent 1ca43ed commit d247c6e

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/conftest.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@ def DEFAULTS():
5151
doubling_time=4.0,
5252
n_days=60,
5353
market_share=0.15,
54+
mitigation_date=datetime(year=2020, month=3, day=28),
5455
relative_contact_rate=0.3,
5556
hospitalized=Disposition(0.025, 7),
5657
icu=Disposition(0.0075, 9),
@@ -65,6 +66,7 @@ def param():
6566
current_hospitalized=100,
6667
doubling_time=6.0,
6768
market_share=0.05,
69+
mitigation_date=datetime(year=2020, month=3, day=28),
6870
relative_contact_rate=0.15,
6971
population=500000,
7072
hospitalized=Disposition(0.05, 7),
@@ -81,6 +83,7 @@ def halving_param():
8183
current_hospitalized=100,
8284
doubling_time=6.0,
8385
market_share=0.05,
86+
mitigation_date=datetime(year=2020, month=3, day=28),
8487
relative_contact_rate=0.7,
8588
population=500000,
8689
hospitalized=Disposition(0.05, 7),

0 commit comments

Comments
 (0)