def test_get_qualify_table(self):
gameday = DBSetup().g62_qualify_finished()
gmw = GamedayModelWrapper(gameday.pk)
expected_qualify_table = get_df_from_json('ts_qualify_table')
assert_frame_equal(gmw.get_qualify_table(),
expected_qualify_table,
check_dtype=False)
class GamedayService:
@classmethod
def create(cls, gameday_pk):
try:
return cls(gameday_pk)
except Gameinfo.DoesNotExist:
return EmptyGamedayService
def __init__(self, pk):
self.gmw = GamedayModelWrapper(pk)
def get_schedule(self):
schedule = self.gmw.get_schedule()
columns = [SCHEDULED, FIELD, HOME, POINTS_HOME, POINTS_AWAY, AWAY, OFFICIALS_NAME, STANDING, STAGE, STATUS]
schedule = schedule[columns]
schedule[OFFICIALS_NAME] = schedule[OFFICIALS_NAME].apply('<i>{}</i>'.format)
schedule[SCHEDULED] = pd.to_datetime(schedule[SCHEDULED], format='%H:%M:%S').dt.strftime('%H:%M')
schedule = schedule.rename(columns=SCHEDULE_TABLE_HEADERS)
return schedule
def get_qualify_table(self):
qualify_table = self.gmw.get_qualify_table()
if qualify_table is '':
return EmptyQualifyTable
qualify_table = qualify_table[[STANDING, TEAM_NAME, POINTS, PF, PA, DIFF]]
qualify_table = qualify_table.rename(columns=TABLE_HEADERS)
return qualify_table
def get_final_table(self):
final_table = self.gmw.get_final_table()
if final_table.empty:
return final_table
final_table = final_table[[DFFL, TEAM_NAME, POINTS, PF, PA, DIFF]]
final_table = final_table.rename(columns=TABLE_HEADERS)
return final_table
def get_games_to_whistle(self, team):
if team == '*':
team = ''
games_to_whistle = self.gmw.get_games_to_whistle(team)
columns = [SCHEDULED, FIELD, OFFICIALS_NAME, STAGE, STANDING, HOME, POINTS_HOME, POINTS_AWAY, AWAY,
STATUS, ID_HOME, ID_AWAY, 'id']
games_to_whistle = games_to_whistle[columns]
games_to_whistle = games_to_whistle.rename(columns={OFFICIALS_NAME: OFFICIALS})
return games_to_whistle
def test_empty_get_qualify_table(self):
gameday = DBSetup().create_main_round_gameday()
gmw = GamedayModelWrapper(gameday.pk)
assert gmw.get_qualify_table() == ''