def test_fg_batting_data_on_active_roster(self, assert_frame_not_equal: Callable[..., bool]) -> None:
data = fg_batting_data(2018, max_results=self.DEFAULT_MAX_RESULTS)
assert data is not None
assert not data.empty
assert len(data.columns) == self.ALL_DATA_COLUMNS_COUNT
assert len(data.index) == self.DEFAULT_MAX_RESULTS
oar_data = fg_batting_data(2018, on_active_roster=True, max_results=self.DEFAULT_MAX_RESULTS)
assert oar_data is not None
assert not oar_data.empty
assert len(oar_data.columns) == self.ALL_DATA_COLUMNS_COUNT
assert len(oar_data.index) == self.DEFAULT_MAX_RESULTS
assert_frame_not_equal(data, oar_data)
def test_batting_stats(self, response_get_monkeypatch: Callable,
test_batting_stats_html: str,
test_batting_stats_result: pd.DataFrame):
season = 2019
query_params = urllib.parse.urlencode(
{
'pos': 'all',
'stats': 'bat',
'lg': 'all',
'qual': 'y',
'type': stat_list_to_str(FangraphsBattingStats.ALL()),
'season': season,
'month': 0,
'season1': season,
'ind': '1',
'team': '',
'rost': '0',
'age': f"{MIN_AGE},{MAX_AGE}",
'filter': '',
'players': '',
'page': f'1_1000000'
},
safe=',')
expected_url = f"{_FG_LEADERS_URL}?{query_params}"
response_get_monkeypatch(test_batting_stats_html, expected_url)
batting_stats_result = fg_batting_data(season).reset_index(drop=True)
pd.testing.assert_frame_equal(batting_stats_result,
test_batting_stats_result)
def test_fg_batting_data_league(self, assert_frame_not_equal: Callable[..., bool]) -> None:
data_al = fg_batting_data(2019, league='AL', max_results=self.DEFAULT_MAX_RESULTS)
assert data_al is not None
assert not data_al.empty
assert len(data_al.columns) == self.ALL_DATA_COLUMNS_COUNT
assert len(data_al.index) == self.DEFAULT_MAX_RESULTS
data_nl = fg_batting_data(2019, league='NL', max_results=self.DEFAULT_MAX_RESULTS)
assert data_nl is not None
assert not data_nl.empty
assert len(data_nl.columns) == self.ALL_DATA_COLUMNS_COUNT
assert len(data_nl.index) == self.DEFAULT_MAX_RESULTS
assert assert_frame_not_equal(data_al, data_nl)
def test_fg_batting_data_qual(self) -> None:
data = fg_batting_data(2019, qual=715, max_results=self.DEFAULT_MAX_RESULTS)
assert data is not None
assert not data.empty
assert len(data.columns) == self.ALL_DATA_COLUMNS_COUNT
assert len(data.index) == 3
def test_fg_batting_position(self, assert_frame_not_equal: Callable[..., bool]) -> None:
data_1b = fg_batting_data(2019, position='1B', max_results=self.DEFAULT_MAX_RESULTS)
assert data_1b is not None
assert not data_1b.empty
assert len(data_1b.columns) == self.ALL_DATA_COLUMNS_COUNT
assert len(data_1b.index) == 10
data_2b = fg_batting_data(2019, position='2B', max_results=self.DEFAULT_MAX_RESULTS)
assert data_2b is not None
assert not data_2b.empty
assert len(data_2b.columns) == self.ALL_DATA_COLUMNS_COUNT
assert len(data_2b.index) == 10
assert_frame_not_equal(data_1b, data_2b)
def test_fg_batting_data_multiple_stat_columns(self) -> None:
data = fg_batting_data(2019, stat_columns=['AB', 'PA'], max_results=self.DEFAULT_MAX_RESULTS)
assert data is not None
assert not data.empty
assert len(data.columns) == 6
assert len(data.index) == self.DEFAULT_MAX_RESULTS
def test_fg_batting_data_end_season_no_split_season(self) -> None:
data = fg_batting_data(2018, end_season=2019, split_seasons=False, max_results=self.DEFAULT_MAX_RESULTS)
assert data is not None
assert not data.empty
assert len(data.columns) == self.ALL_DATA_COLUMNS_COUNT - 1
assert 'Season' not in data.columns
assert len(data.index) == self.DEFAULT_MAX_RESULTS
def test_fg_batting_data_max_results(self) -> None:
season = 2019
data = fg_batting_data(season)
assert data is not None
assert not data.empty
assert len(data.columns) == self.ALL_DATA_COLUMNS_COUNT
assert len(data.index) == 135
def test_fg_batting_team(self, assert_frame_not_equal: Callable[..., bool]) -> None:
data_1 = fg_batting_data(2019, team='1', max_results=self.DEFAULT_MAX_RESULTS)
assert data_1 is not None
assert not data_1.empty
assert len(data_1.columns) == self.ALL_DATA_COLUMNS_COUNT - 1
assert 'Team' not in data_1.columns
assert len(data_1.index) == 4
data_2 = fg_batting_data(2019, team='2', max_results=self.DEFAULT_MAX_RESULTS)
assert data_2 is not None
assert not data_2.empty
assert len(data_2.columns) == self.ALL_DATA_COLUMNS_COUNT - 1
assert 'Team' not in data_2.columns
assert len(data_2.index) == 4
assert_frame_not_equal(data_1, data_2)
def test_fg_batting_data_single_stat_columns(self):
data = fg_batting_data(2019,
stat_columns='AB',
max_results=self.DEFAULT_MAX_RESULTS)
assert data is not None
assert not data.empty
assert len(data.columns) == 4
assert len(data.index) == self.DEFAULT_MAX_RESULTS
def test_fg_batting_data_end_season(self):
data = fg_batting_data(2018,
end_season=2019,
max_results=self.DEFAULT_MAX_RESULTS)
assert data is not None
assert not data.empty
assert len(data.columns) == self.ALL_DATA_COLUMNS_COUNT
assert len(data.index) == self.DEFAULT_MAX_RESULTS
def test_fg_batting_minimum_age(self):
data = fg_batting_data(2019,
minimum_age=39,
max_results=self.DEFAULT_MAX_RESULTS)
assert data is not None
assert not data.empty
assert len(data.columns) == self.ALL_DATA_COLUMNS_COUNT
assert len(data.index) == 1
def test_fg_batting_data(self) -> None:
season = 2019
data = fg_batting_data(season, max_results=self.DEFAULT_MAX_RESULTS)
assert data is not None
assert not data.empty
assert len(data.columns) == self.ALL_DATA_COLUMNS_COUNT
assert len(data.index) == self.DEFAULT_MAX_RESULTS
seasons = list(set(data['Season']))
assert len(seasons) == 1
assert seasons[0] == season
def test_fg_batting_data_future_season(self) -> None:
season = 3000
with pytest.raises(ValueError):
fg_batting_data(season, max_results=self.DEFAULT_MAX_RESULTS)
