def test_estimate(self, jhu_data, population_data, country):
warnings.simplefilter("ignore", category=UserWarning)
# Setting
snl = Scenario(jhu_data, population_data, country)
snl.first_date = "01Apr2020"
snl.last_date = "01Aug2020"
with pytest.raises(ValueError):
snl.estimate(SIR)
snl.trend(include_init_phase=True, show_figure=False)
snl.disable(phases=["0th"])
with pytest.raises(AttributeError):
snl.estimate_history(phase="1th")
# Parameter estimation
with pytest.raises(KeyError):
snl.estimate(SIR, phases=["30th"])
with pytest.raises(ValueError):
snl.estimate(model=SIR, tau=1440)
snl.enable(phases=["0th"])
with pytest.raises(TypeError):
snl.estimate(model=SIR, phases="1st")
with pytest.raises(ValueError):
snl.estimate(model=SIR, phases=["0th"])
snl.clear(include_past=True)
snl.trend(show_figure=False)
snl.estimate(SIR)
# Estimation history
snl.estimate_history(phase="1st")
# Estimation accuracy
snl.estimate_accuracy(phase="1st")
# Get a value
snl.get(Term.RT)
with pytest.raises(KeyError):
snl.get("feeling")
def test_trend(self, jhu_data, population_data, country):
# Setting
snl = Scenario(jhu_data, population_data, country)
snl.first_date = "01Apr2020"
snl.last_date = "01Aug2020"
# Deprecated
warnings.simplefilter("error")
with pytest.raises(DeprecationWarning):
snl.trend(include_init_phase=False, show_figure=False)
warnings.simplefilter("ignore")
snl.trend(include_init_phase=False, show_figure=False)
# S-R trend analysis
snl.trend(show_figure=False)
assert snl["Main"]
with pytest.raises(ValueError):
snl.trend(show_figure=False, n_points=3)
# Disable/enable
length = len(snl["Main"])
snl.disable(phases=["0th"], name="Main")
assert len(snl["Main"]) == length - 1
snl.enable(phases=["0th"], name="Main")
assert len(snl["Main"]) == length
with pytest.raises(TypeError):
snl.enable(phases="0th", name="Main")
with pytest.raises(TypeError):
snl.disable(phases="1st", name="Main")
def test_trend(self, jhu_data, population_data, country):
# Setting
snl = Scenario(jhu_data, population_data, country)
snl.first_date = "01Apr2020"
snl.last_date = "01Aug2020"
snl.trend(show_figure=False)
assert snl["Main"]
with pytest.raises(ValueError):
snl.trend(show_figure=False, n_points=3)
# Disable/enable
length = len(snl["Main"])
snl.disable(phases=["0th"], name="Main")
assert len(snl["Main"]) == length - 1
snl.enable(phases=["0th"], name="Main")
assert len(snl["Main"]) == length
with pytest.raises(TypeError):
snl.enable(phases="0th", name="Main")
with pytest.raises(TypeError):
snl.disable(phases="1st", name="Main")