def test_away_team_data_field_name(self): formatter = RowFormatter(data_field_names=["away_team"]) self.assertEqual( formatter.format(row_data={"away_team": Team.BOSTON_CELTICS}), {"away_team": Team.BOSTON_CELTICS.value})
def test_multiple_positions_data_field_name(self): formatter = RowFormatter(data_field_names=["positions"]) self.assertEqual( formatter.format(row_data={ "positions": [Position.POINT_GUARD, Position.SHOOTING_GUARD] }), {"positions": "POINT GUARD-SHOOTING GUARD"})
def test_non_team_outcome_location_position_data_field_name(self): formatter = RowFormatter(data_field_names=["name"]) self.assertEqual(formatter.format(row_data={"name": "jaebaebae"}), {"name": "jaebaebae"})
def test_single_position_data_field_name(self): formatter = RowFormatter(data_field_names=["positions"]) self.assertEqual( formatter.format(row_data={"positions": [Position.POINT_GUARD]}), {"positions": "POINT GUARD"})
def test_empty_positions_data_field_name(self): formatter = RowFormatter(data_field_names=["positions"]) self.assertEqual(formatter.format(row_data={"positions": []}), {"positions": ""})
def test_outcome_data_field_name(self): formatter = RowFormatter(data_field_names=["outcome"]) self.assertEqual(formatter.format(row_data={"outcome": Outcome.WIN}), {"outcome": Outcome.WIN.value})
def test_opponent_data_field_name(self): formatter = RowFormatter(data_field_names=["opponent"]) self.assertEqual( formatter.format(row_data={"opponent": Team.BOSTON_CELTICS}), {"opponent": Team.BOSTON_CELTICS.value})
def test_location_data_field_name(self): formatter = RowFormatter(data_field_names=["location"]) self.assertEqual( formatter.format(row_data={"location": Location.HOME}), {"location": Location.HOME.value})