|
51 | 51 | )
|
52 | 52 |
|
53 | 53 | PARAM = Parameters(
|
| 54 | + current_date=datetime(year=2020, month=3, day=28), |
54 | 55 | current_hospitalized=100,
|
55 | 56 | doubling_time=6.0,
|
56 | 57 | market_share=0.05,
|
|
63 | 64 | )
|
64 | 65 |
|
65 | 66 | HALVING_PARAM = Parameters(
|
| 67 | + current_date=datetime(year=2020, month=3, day=28), |
66 | 68 | current_hospitalized=100,
|
67 | 69 | doubling_time=6.0,
|
68 | 70 | market_share=0.05,
|
@@ -287,19 +289,19 @@ def test_model_raw_start():
|
287 | 289 |
|
288 | 290 | # test the things n_days creates, which in turn tests sim_sir, sir, and get_dispositions
|
289 | 291 |
|
290 |
| - # print('n_days: %s; i_day: %s' % (param.n_days, model.i_day)) |
| 292 | + print('n_days: %s; i_day: %s' % (param.n_days, model.i_day)) |
291 | 293 | assert len(raw_df) == (len(np.arange(-model.i_day, param.n_days + 1))) == 104
|
292 | 294 |
|
293 | 295 | first = raw_df.iloc[0, :]
|
294 | 296 | second = raw_df.iloc[1, :]
|
295 | 297 |
|
296 | 298 | assert first.susceptible == 499600.0
|
297 | 299 | assert round(second.infected, 0) == 449.0
|
298 |
| - assert list(model.dispositions_df.iloc[0, :]) == [-43, date(year=2020, month=2, day=15), 1.0, 0.4, 0.2] |
| 300 | + assert list(model.dispositions_df.iloc[0, :]) == [-43, date(year=2020, month=2, day=14), 1.0, 0.4, 0.2] |
299 | 301 | assert round(raw_df.recovered[30], 0) == 7083.0
|
300 | 302 |
|
301 | 303 | d, dt, s, i, r = list(model.dispositions_df.iloc[60, :])
|
302 |
| - assert dt == date(year=2020, month=4, day=15) |
| 304 | + assert dt == date(year=2020, month=4, day=14) |
303 | 305 | assert [round(v, 0) for v in (d, s, i, r)] == [17, 549.0, 220.0, 110.0]
|
304 | 306 |
|
305 | 307 |
|
|
0 commit comments