def test_get_country_data():
# Germany
cases, deaths, region_label = c.get_country_data(country="Germany",
subregion="SK Hamburg")
print(f"region_label = {region_label}")
print(f"deaths = {type(deaths)}")
print(f"empty")
assert cases.name == 'Germany-SK Hamburg cases'
assert deaths.name == 'Germany-SK Hamburg deaths'
assert region_label == 'Germany-SK Hamburg'
c2, d2, region_label = c.get_country_data(country="United Kingdom")
assert c2.name == "United Kingdom cases"
assert d2.name == "United Kingdom deaths"
assert region_label == "United Kingdom"
def test_get_country_data():
# Germany
cases, deaths = c.get_country_data(country="Germany", region="Bayern")
assert isinstance(deaths, pd.Series)
assert cases.name == 'Germany-Bayern cases'
assert deaths.name == 'Germany-Bayern deaths'
cases, deaths = c.get_country_data(country="Germany",
subregion="SK Hamburg")
assert isinstance(deaths, pd.Series)
assert cases.name == 'Germany-SK Hamburg cases'
assert deaths.name == 'Germany-SK Hamburg deaths'
c2, d2 = c.get_country_data(country="United Kingdom")
assert c2.name == "United Kingdom cases"
assert d2.name == "United Kingdom deaths"
def init_metadata(self):
cases, _ = oscovida.get_country_data("Hungary", region=self.region)
self._init_metadata(
meta={
"source": "https://github.com/sanbrock/covid19",
"category": self.category,
"max-deaths": None,
"max-cases": int(cases[-1]),
"region": self.region,
"subregion": str(None),
"one-line-summary":
f"Hungary: {self.region}", # used as title in table
"cases-last-week": int(oscovida.get_cases_last_week(cases)),
})
def init_metadata(self):
cases, deaths = oscovida.get_country_data(self.country)
one_line_summary = f"{self.country}"
self._init_metadata(
meta={
"source": "CSSE Johns Hopkins",
"category": self.category,
"max-deaths": int(deaths[-1]),
"max-cases": int(cases[-1]),
"region": str(None),
"subregion": str(None),
"one-line-summary": one_line_summary, # used as title in table
"cases-last-week": int(oscovida.get_cases_last_week(cases)),
})
def init_metadata(self):
cases, deaths = oscovida.get_country_data("Germany",
subregion=self.subregion)
one_line_summary = f"Germany: {self.region} : {self.subregion}"
self._init_metadata(
meta={
"source": "Robert Koch Institute",
"category": self.category,
"max-deaths": int(deaths[-1]),
"max-cases": int(cases[-1]),
"region": self.region,
"subregion": self.subregion,
"one-line-summary": one_line_summary, # used as title in table
"cases-last-week": int(oscovida.get_cases_last_week(cases)),
})