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)