def test_germany_overview():
axes, cases, deaths = c.overview(country="Germany", region="Hamburg")
assert cases.name == 'Germany-Hamburg cases'
assert_oscovida_object(axes, cases, deaths)
axes, cases, deaths = c.overview(country="Germany",
subregion="LK Pinneberg")
assert deaths.name == 'Germany-LK Pinneberg deaths'
assert_oscovida_object(axes, cases, deaths)
axes, cases, deaths = c.overview(country="Germany", subregion="SK Kassel")
assert cases.name == 'Germany-SK Kassel cases'
assert deaths.name == 'Germany-SK Kassel deaths'
assert_oscovida_object(axes, cases, deaths)
axes, cases, deaths = c.overview(country="Germany",
subregion="StadtRegion Aachen")
assert cases.name == 'Germany-StadtRegion Aachen cases'
assert_oscovida_object(axes, cases, deaths)
axes, cases, deaths = c.overview(country="Germany",
subregion="Region Hannover")
assert cases.name == 'Germany-Region Hannover cases'
assert deaths.name == 'Germany-Region Hannover deaths'
assert_oscovida_object(axes, cases, deaths)
def test_world_overview(n=10):
countries = get_country_list()
start_time = time.time()
for i, country in enumerate(countries[0:n]):
print(f"Processing {i+1:3}/{len(countries)} [{time.time()-start_time:4.0f}s] {country}")
overview(country=country)
matplotlib.pyplot.close('all')
def test_germany_overview(n=10):
"""Test n countries """
subregions = get_germany_subregion_list()[0:n]
start_time = time.time()
for i, subregion in enumerate(subregions):
print(f"Processing {i+1:3}/{len(subregions)} [{time.time()-start_time:4.0f}s] {subregion}")
overview(country="Germany", subregion=subregion)
matplotlib.pyplot.close('all')
def test_Hungary_overview():
axes, cases, deaths = c.overview(country="Hungary", region="Baranya")
assert cases.name == 'Hungary-Baranya cases'
assert deaths is None
isinstance(cases, pd.Series)
isinstance(deaths, type(None))
def test_US_overview():
axes, cases, deaths = c.overview(country="US", region="New Jersey")
assert cases.name == 'US-New Jersey cases'
assert deaths.name == 'US-New Jersey deaths'
isinstance(deaths, pd.core.series.Series)
isinstance(deaths, pd.core.series.Series)
def test_overview():
axes, cases, deaths = c.overview("China")
assert cases.name == 'China cases'
assert deaths.name == 'China deaths'
isinstance(deaths, pd.core.series.Series)
isinstance(deaths, pd.core.series.Series)
def test_germany_overview(n=10):
"""Test n countries """
subregions = get_germany_subregion_list()[0:n]
# data cleaning: on 13 April, we had a Landkreis "LK Göttingen (alt)"
# with only one data point. This causes plots to fail, because there
# is nothing to plot, and then the legend() command failed.
# We assume that the RKI labels unusual data with '(alt)', and remove those.
alt_data_sets = [x for x in subregions if "(alt)" in x.lower()]
if len(alt_data_sets) > 0:
print(f"Removing datasets label with '(alt)': {alt_data_sets}")
for alt in alt_data_sets:
c, d = germany_get_region(landkreis=alt)
print(f" removed: {alt} : len(cases)={len(c)}, len(deaths)={len(d)}")
# subregions = [x for x in subregions if not "(alt)" in x.lower()]
start_time = time.time()
for i, subregion in enumerate(subregions):
print(f"Processing {i+1:3}/{len(subregions)} [{time.time()-start_time:4.0f}s] {subregion}")
overview(country="Germany", subregion=subregion)
matplotlib.pyplot.close('all')
def test_overview():
axes, cases, deaths = c.overview("China")
assert cases.name == 'China cases'
assert deaths.name == 'China deaths'
assert_oscovida_object(axes, cases, deaths)
assert_oscovida_object(*c.overview("Germany", weeks=8))
assert_oscovida_object(
*c.overview("Russia", dates="2020-05-30:2020-06-15"))
with pytest.raises(ValueError):
c.overview("Argentina", weeks=8, dates="2020-05-30:2020-06-15")
days = 10
dates = pd.date_range("2020-03-01", periods=days, freq='D')
data1 = np.exp(np.linspace(1, 15, days))
data2 = np.exp(np.linspace(1, 5, days))
cases = pd.Series(data1, index=pd.DatetimeIndex(dates))
deaths = pd.Series(data2, index=pd.DatetimeIndex(dates))
assert_oscovida_object(*c.overview("Narnia", data=(cases, deaths)))
def test_US_overview():
axes, cases, deaths = c.overview(country="US", region="New Jersey")
assert cases.name == 'US-New Jersey cases'
assert deaths.name == 'US-New Jersey deaths'
assert_oscovida_object(axes, cases, deaths)
