def test_games(self): sched = Schedule('1/1/2015') games = sched.get_games() size = len(games) exp_size = 1230 self.assertEqual(size, exp_size) cols = len(games.columns) exp_cols = 40 + 4 self.assertEqual(cols, exp_cols)