def test_policy_measures(self, jhu_data, population_data, oxcgrt_data):
warnings.filterwarnings("ignore", category=UserWarning)
# Create instance
analyser = PolicyMeasures(jhu_data,
population_data,
oxcgrt_data,
tau=360)
# List of countries
assert isinstance(analyser.countries, list)
# Return Scenario class
assert isinstance(analyser.scenario("Japan"), Scenario)
with pytest.raises(KeyError):
analyser.scenario("Moon")
assert isinstance(analyser.countries, list)
# S-R trend analysis
analyser.trend()
min_len = max(analyser.phase_len().keys())
analyser.trend(min_len=min_len)
# Summarize
assert isinstance(analyser.summary(), pd.DataFrame)
with pytest.raises(TypeError):
analyser.summary(countries="Poland")
# Phase length
phase_len_dict = analyser.phase_len()
assert isinstance(phase_len_dict, dict)
assert isinstance(phase_len_dict[min_len], list)
# Parameter estimation
analyser.estimate(SIRF)
assert isinstance(analyser.summary(), pd.DataFrame)
# Parameter history of Rt
df = analyser.param_history("Rt", roll_window=None)
assert isinstance(df, pd.DataFrame)
# Parameter history of rho
df = analyser.param_history("rho", roll_window=14, show_figure=False)
assert isinstance(df, pd.DataFrame)
def test_start(self, jhu_data, population_data, oxcgrt_data):
warnings.simplefilter("ignore", category=UserWarning)
# Create instance
analyser = PolicyMeasures(jhu_data, population_data, oxcgrt_data, tau=360)
# List of countries
assert isinstance(analyser.countries, list)
# Return Scenario class
assert isinstance(analyser.scenario("Japan"), Scenario)
with pytest.raises(KeyError):
analyser.scenario("Moon")