def test_fg_fielding_team(
self, assert_frame_not_equal: Callable[..., bool]) -> None:
data_1 = fg_fielding_data(2019,
team='1',
qual=750,
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) == 5
data_2 = fg_fielding_data(2019,
team='2',
qual=750,
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) == 3
assert_frame_not_equal(data_1, data_2)
def test_fg_fielding_data_qual(self) -> None:
data = fg_fielding_data(2019,
qual=1400,
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) == 2
def test_fg_fielding_data_multiple_stat_columns(self) -> None:
data = fg_fielding_data(2019,
stat_columns=['DEF', 'UZR'],
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_fielding_data_max_results(self) -> None:
season = 2019
data = fg_fielding_data(season)
assert data is not None
assert not data.empty
assert len(data.columns) == self.ALL_DATA_COLUMNS_COUNT
assert len(data.index) == 107
def test_fg_fielding_data_on_active_roster(
self, assert_frame_not_equal: Callable[..., bool]) -> None:
data = fg_fielding_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_fielding_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_fg_fielding_data_end_season_no_split_season(self) -> None:
data = fg_fielding_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_fielding_data_league(
self, assert_frame_not_equal: Callable[..., bool]) -> None:
data_al = fg_fielding_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_fielding_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_fielding_position(
self, assert_frame_not_equal: Callable[..., bool]) -> None:
data_1b = fg_fielding_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_fielding_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_fielding_data(self) -> None:
season = 2019
data = fg_fielding_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_fielding_data_future_season(self) -> None:
season = 3000
with pytest.raises(ValueError):
fg_fielding_data(season, max_results=self.DEFAULT_MAX_RESULTS)
