def test_create_table(self):
PATH = Path('tests/fixtures/table_columns.json')
columns = load_json(PATH)
loader = NflLoader()
season, week, team = (2018, 1, 'CAR')
table = loader.get_game_table(season, week, team, update_schedule=False, new=True)
self.assertListEqual(columns, list(table.columns))
def test_add_fpts(self):
PATH = Path('tests/fixtures/test_fpts.csv')
df = pd.read_csv(PATH)
loader = NflLoader()
df = loader.add_fpts(df)
fpts_calculated = list(df['fpts'])
fpts_soll = list(df['soll_fpts'])
self.assertListEqual(fpts_calculated, fpts_soll)
def get_stats_from_current_season(weeks: list):
season = 2019
loader = NflLoader(season)
weekdata = []
for week in weeks:
weekdata.append(loader.get_weektable(week))
df = pd.concat(weekdata, sort=False)
df.to_csv(loader.datapath.parent / f"{season}_REG.csv")
print(df.head())
def test_get_game_eid(self):
nloader = NflLoader()
eid = nloader.get_game_eid("test", "test_sched", "test_team", update_schedule=False)
self.assertEqual(eid, "test_eid")
def test_get_game_stats(self):
nloader = NflLoader()
nloader.datapath = Path('tests/database')
stats = nloader.get_game_stats("test_eid")
self.assertEqual(stats['test_eid']['home']['abbr'], "test_team")
def get_pre_season(season: int):
loader = NflLoader(season, seasontype='PRE')
stable = loader.get_seasontable()
return stable
def get_stats():
seasons = (2014, 2015, 2016)
for season in seasons:
nl = NflLoader(season, save=True)
_ = nl.get_seasontable()