def test_analysis(self, jhu_data, population_data, oxcgrt_data):
warnings.simplefilter("ignore", category=UserWarning)
warnings.simplefilter("error", category=RuntimeWarning)
# Create instance
analyser = PolicyMeasures(jhu_data,
population_data,
oxcgrt_data,
tau=360)
# S-R trend analysis
analyser.trend()
# Phase length
phase_len_dict = analyser.phase_len()
assert isinstance(phase_len_dict, dict)
assert isinstance(
max(phase_len_dict.items(), key=lambda x: x[0])[1], list)
# Select two countries
phase_len_dict = analyser.phase_len()
countries_all = [
country for (_, countries) in sorted(phase_len_dict.items())
for country in countries
]
analyser.countries = countries_all[:2]
# Parameter estimation
with pytest.raises(ValueError):
analyser.track()
analyser.estimate(SIRF, timeout=1, timeout_iteration=1)
assert isinstance(analyser.summary(), pd.DataFrame)
# Parameter history of Rt
with pytest.raises(KeyError):
df = analyser.history("Temperature", roll_window=None)
df = analyser.history("Rt", roll_window=None)
assert isinstance(df, pd.DataFrame)
# Parameter history of rho
df = analyser.history("rho", roll_window=14, show_figure=False)
assert isinstance(df, pd.DataFrame)
# Summarize
assert isinstance(analyser.summary(), pd.DataFrame)
with pytest.raises(TypeError):
analyser.summary(countries="Poland")
def test_policy_measures(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")
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
with pytest.raises(ValueError):
analyser.track()
analyser.estimate(SIRF)
assert isinstance(analyser.summary(), pd.DataFrame)
# Parameter history of Rt
with pytest.raises(KeyError):
df = analyser.history("Temperature", roll_window=None)
df = analyser.history("Rt", roll_window=None)
assert isinstance(df, pd.DataFrame)
# Parameter history of rho
df = analyser.history("rho", roll_window=14, show_figure=False)
assert isinstance(df, pd.DataFrame)