def test_joindataset_date(self):
newYorkDataExtraction = getData(self.nytdata)
johnHopkinsDataExtraction = getData(self.jphdata)
NYTData = transformNewYorkTimesData(newYorkDataExtraction)
JPHData = transformJohnHopkinsData(johnHopkinsDataExtraction)
result = transformJoinData(NYTData, JPHData)
result_date = str(result.datetime.iloc[0])
self.assertEqual(result_date, '2020-08-12 00:00:00')
def test_joindataset_headers(self):
newYorkDataExtraction = getData(self.nytdata)
johnHopkinsDataExtraction = getData(self.jphdata)
NYTData = transformNewYorkTimesData(newYorkDataExtraction)
JPHData = transformJohnHopkinsData(johnHopkinsDataExtraction)
result = transformJoinData(NYTData, JPHData)
result_headers = list(result.columns)
expected_headers = ['datetime', 'cases', 'deaths', 'Recovered']
self.assertEqual(result_headers, expected_headers)
def test_joindataset_int(self):
newYorkDataExtraction = getData(self.nytdata)
johnHopkinsDataExtraction = getData(self.jphdata)
NYTData = transformNewYorkTimesData(newYorkDataExtraction)
JPHData = transformJohnHopkinsData(johnHopkinsDataExtraction)
result = transformJoinData(NYTData, JPHData)
result_cases_int = result.cases.iloc[0]
result_deaths_int = result.deaths.iloc[0]
result_Recocered_int = result.Recovered.iloc[0]
self.assertGreaterEqual(result_cases_int, 0)
self.assertGreaterEqual(result_deaths_int, 0)
self.assertGreaterEqual(result_Recocered_int, 0)
def test_nytdata_int(self):
result = transformNewYorkTimesData(getData(self.nytdata))
result_cases_int = result.cases.iloc[0]
result_deaths_int = result.deaths.iloc[1]
self.assertGreaterEqual(result_cases_int, 0)
self.assertGreaterEqual(result_deaths_int, 0)
def test_nytdata(self):
result = transformNewYorkTimesData(getData(self.nytdata))
result_date = str(result.datetime.iloc[1])
self.assertEqual(result_date, '2020-12-12 00:00:00')
def test_nytdata_headers(self):
result = transformNewYorkTimesData(getData(self.nytdata))
result_headers = list(result.columns)
result_df = ['datetime', 'cases', 'deaths']
self.assertEqual(result_headers, result_df)