def test_get_game_shifts(self):
try:
res = api.get_game_shifts(self.game_id)
self.assertIsInstance(res, DataFrame)
self.assertGreater(
res.shape[0],
200) # just a rough ball-park check for the number of events
self.assertGreater(
res.shape[1],
3) # another rough ball-park check for the number of columns
except Exception as e:
self.fail(
"api.get_game_shifts() returned unexpected exception: {}".
format(str(e)))
def _scrape_game(output_format: OutputFormat, game_ids):
output_format.echo(get_game_shifts(*game_ids))