def test_error(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)
# Register countries
with pytest.raises(KeyError):
analyser.countries = ["Moon"]
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")