def setup_method(self, *args, **kwargs):
self.results = {
'game': 2,
'boxscore_index': '2017-09-09-michigan',
'date': 'Sep 9, 2017',
'time': '12:00 PM',
'day_of_week': 'Sat',
'datetime': datetime(2017, 9, 9, 12, 0),
'location': HOME,
'rank': 8,
'opponent_abbr': 'cincinnati',
'opponent_name': 'Cincinnati',
'opponent_rank': None,
'opponent_conference': 'American',
'result': WIN,
'points_for': 36,
'points_against': 14,
'wins': 2,
'losses': 0,
'streak': 'W 2'
}
flexmock(utils) \
.should_receive('_todays_date') \
.and_return(MockDateTime(YEAR, MONTH))
flexmock(Boxscore) \
.should_receive('_parse_game_data') \
.and_return(None)
flexmock(Boxscore) \
.should_receive('dataframe') \
.and_return(pd.DataFrame([{'key': 'value'}]))
self.schedule = Schedule('MICHIGAN')
def test_empty_page_return_no_games(self):
flexmock(utils) \
.should_receive('_no_data_found') \
.once()
flexmock(utils) \
.should_receive('_get_stats_table') \
.and_return(None)
schedule = Schedule('MICHIGAN')
assert len(schedule) == 0
def test_no_dataframes_extended_returns_none(self):
flexmock(Schedule) \
.should_receive('_pull_schedule') \
.and_return(None)
schedule = Schedule('PURDUE')
fake_game = flexmock(dataframe_extended=None)
fake_games = PropertyMock(return_value=fake_game)
type(schedule).__iter__ = fake_games
assert schedule.dataframe_extended is None
def test_invalid_default_year_reverts_to_previous_year(self,
*args,
**kwargs):
results = {
'game': 2,
'boxscore_index': '2017-09-09-michigan',
'date': 'Sep 9, 2017',
'time': '12:00 PM',
'day_of_week': 'Sat',
'datetime': datetime(2017, 9, 9, 12, 0),
'location': HOME,
'rank': 8,
'opponent_abbr': 'cincinnati',
'opponent_name': 'Cincinnati',
'opponent_rank': None,
'opponent_conference': 'American',
'result': WIN,
'points_for': 36,
'points_against': 14,
'wins': 2,
'losses': 0,
'streak': 'W 2'
}
flexmock(Boxscore) \
.should_receive('_parse_game_data') \
.and_return(None)
flexmock(Boxscore) \
.should_receive('dataframe') \
.and_return(pd.DataFrame([{'key': 'value'}]))
flexmock(utils) \
.should_receive('_find_year_for_season') \
.and_return(2018)
schedule = Schedule('MICHIGAN')
for attribute, value in results.items():
assert getattr(schedule[1], attribute) == value
class TestNCAAFSchedule:
@mock.patch('requests.get', side_effect=mock_pyquery)
def setup_method(self, *args, **kwargs):
self.results = {
'game': 2,
'boxscore_index': '2017-09-09-michigan',
'date': 'Sep 9, 2017',
'time': '12:00 PM',
'day_of_week': 'Sat',
'datetime': datetime(2017, 9, 9, 12, 0),
'location': HOME,
'rank': 8,
'opponent_abbr': 'cincinnati',
'opponent_name': 'Cincinnati',
'opponent_rank': None,
'opponent_conference': 'American',
'result': WIN,
'points_for': 36,
'points_against': 14,
'wins': 2,
'losses': 0,
'streak': 'W 2'
}
flexmock(utils) \
.should_receive('_todays_date') \
.and_return(MockDateTime(YEAR, MONTH))
flexmock(Boxscore) \
.should_receive('_parse_game_data') \
.and_return(None)
flexmock(Boxscore) \
.should_receive('dataframe') \
.and_return(pd.DataFrame([{'key': 'value'}]))
self.schedule = Schedule('MICHIGAN')
def test_ncaaf_schedule_returns_correct_number_of_games(self):
assert len(self.schedule) == NUM_GAMES_IN_SCHEDULE
def test_ncaaf_schedule_returns_requested_match_from_index(self):
match_two = self.schedule[1]
for attribute, value in self.results.items():
assert getattr(match_two, attribute) == value
def test_ncaaf_schedule_returns_requested_match_from_date(self):
match_two = self.schedule(datetime(2017, 9, 9))
for attribute, value in self.results.items():
assert getattr(match_two, attribute) == value
def test_ncaaf_schedule_dataframe_returns_dataframe(self):
df = pd.DataFrame([self.results], index=['MICHIGAN'])
match_two = self.schedule[1]
# Pandas doesn't natively allow comparisons of DataFrames.
# Concatenating the two DataFrames (the one generated during the test
# and the expected one above) and dropping duplicate rows leaves only
# the rows that are unique between the two frames. This allows a quick
# check of the DataFrame to see if it is empty - if so, all rows are
# duplicates, and they are equal.
frames = [df, match_two.dataframe]
df1 = pd.concat(frames).drop_duplicates(keep=False)
assert df1.empty
def test_ncaaf_schedule_dataframe_extended_returns_dataframe(self):
df = pd.DataFrame([{'key': 'value'}])
result = self.schedule[1].dataframe_extended
frames = [df, result]
df1 = pd.concat(frames).drop_duplicates(keep=False)
assert df1.empty
def test_ncaaf_schedule_all_dataframe_returns_dataframe(self):
result = self.schedule.dataframe.drop_duplicates(keep=False)
assert len(result) == NUM_GAMES_IN_SCHEDULE
assert set(result.columns.values) == set(self.results.keys())
def test_ncaaf_schedule_all_dataframe_extended_returns_dataframe(self):
result = self.schedule.dataframe_extended
assert len(result) == NUM_GAMES_IN_SCHEDULE
def test_no_games_for_date_raises_value_error(self):
with pytest.raises(ValueError):
self.schedule(datetime.now())
def test_empty_page_return_no_games(self):
flexmock(utils) \
.should_receive('_no_data_found') \
.once()
flexmock(utils) \
.should_receive('_get_stats_table') \
.and_return(None)
schedule = Schedule('MICHIGAN')
assert len(schedule) == 0
def test_game_string_representation(self):
game = self.schedule[0]
assert game.__repr__() == 'Sep 2, 2017 - florida'
def test_schedule_string_representation(self):
expected = """Sep 2, 2017 - florida
Sep 9, 2017 - cincinnati
Sep 16, 2017 - air-force
Sep 23, 2017 - purdue
Oct 7, 2017 - michigan-state
Oct 14, 2017 - indiana
Oct 21, 2017 - penn-state
Oct 28, 2017 - rutgers
Nov 4, 2017 - minnesota
Nov 11, 2017 - maryland
Nov 18, 2017 - wisconsin
Nov 25, 2017 - ohio-state
Jan 1, 2018 - south-carolina"""
assert self.schedule.__repr__() == expected