def test_get_cases_last_week():
index = pd.date_range(start='1/1/2018', end='1/08/2018', freq='D')
z = pd.Series(np.zeros(shape=index.shape), index=index)
assert c.get_cases_last_week(z) == 0
index = pd.date_range(start='1/1/2018', end='1/08/2018', freq='D')
z = pd.Series(np.ones(shape=index.shape), index=index)
assert c.get_cases_last_week(z) == 0
assert c.get_cases_last_week(z.cumsum()) == 7
cases, deaths = mock_get_country_data_johns_hopkins(country="China")
assert c.get_cases_last_week(cases) == 430
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_region_US(self.region)
one_line_summary = f"US: {self.region}"
self._init_metadata(
meta={
"source": "Johns Hopkins University CSSE",
"category": self.category,
"max-deaths": int(deaths[-1]),
"max-cases": int(cases[-1]),
"region": self.region,
"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(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)),
})
