def test_get_shots_has_2017_actual_data(self):
df = shotlink.get_shots([2017], DATA_PATH)
self.assertEqual(1214437, len(df))
def test_prepare_data_converts_date_field(self):
df = shotlink.get_shots([2017], TEST_DATA_PATH)
df = shotlink.prepare_shots(df)
self.assertEqual('datetime64[ns]', str(df['Date'].dtype))
def test_get_shots_has_explicit_data_types(self):
df = shotlink.get_shots([2017], TEST_DATA_PATH)
for field_name, dtype in shotlink.shot_dtypes.items():
self.assertEqual(dtype, df.dtypes[field_name])
def test_get_shots_updates_field_names(self):
df = shotlink.get_shots([2017], TEST_DATA_PATH)
self.assertListEqual(shotlink.shot_fields, df.columns.values.tolist())
def test_get_shots_has_two_years_worth_of_data(self):
df = shotlink.get_shots([2008, 2009], TEST_DATA_PATH)
self.assertEqual(999 + 999, len(df))
def test_get_shots_has_2017_data(self):
df = shotlink.get_shots([2017], TEST_DATA_PATH)
self.assertEqual(999, len(df))