def test_sim_sir(): """ Rounding to move fast past decimal place issues """ raw_df = sim_sir_df( 5, # s 6, # i 7, # r 0.1, # gamma 0, # i_day 0.1, # beta1 40, # n_days1 ) first = raw_df.iloc[0, :] last = raw_df.iloc[-1, :] assert round(first.susceptible, 0) == 5 assert round(first.infected, 2) == 6 assert round(first.recovered, 0) == 7 assert round(last.susceptible, 2) == 0 assert round(last.infected, 2) == 0.18 assert round(last.recovered, 2) == 17.82 assert isinstance(raw_df, pd.DataFrame)
def test_sim_sir(): """ Rounding to move fast past decimal place issues """ raw_df = sim_sir_df(5, 6, 7, 0.1, 0.1, 40) first = raw_df.iloc[0, :] last = raw_df.iloc[-1, :] assert round(first.susceptible, 0) == 5 assert round(first.infected, 2) == 6 assert round(first.recovered, 0) == 7 assert round(last.susceptible, 2) == 0 assert round(last.infected, 2) == 0.18 assert round(last.recovered, 2) == 17.82 assert isinstance(raw_df, pd.DataFrame)