|
|
import os |
|
|
from datetime import datetime |
|
|
|
|
|
import pandas as pd |
|
|
from pytest import fixture |
|
|
from socceraction.data.opta import ( |
|
|
OptaGameSchema, |
|
|
OptaPlayerSchema, |
|
|
OptaTeamSchema, |
|
|
) |
|
|
from socceraction.data.opta.parsers import F9JSONParser |
|
|
|
|
|
|
|
|
@fixture() |
|
|
def f9json_parser() -> F9JSONParser: |
|
|
path = os.path.join( |
|
|
os.path.dirname(__file__), |
|
|
os.pardir, |
|
|
os.pardir, |
|
|
os.pardir, |
|
|
"datasets", |
|
|
"opta", |
|
|
"match-2017-8-918893.json", |
|
|
) |
|
|
return F9JSONParser(str(path)) |
|
|
|
|
|
|
|
|
def test_extract_games(f9json_parser: F9JSONParser) -> None: |
|
|
games = f9json_parser.extract_games() |
|
|
assert len(games) == 1 |
|
|
assert games[918893] == { |
|
|
"game_id": 918893, |
|
|
"season_id": 2017, |
|
|
"competition_id": 8, |
|
|
"game_day": 1, |
|
|
"game_date": datetime(2017, 8, 11, 18, 45), |
|
|
"home_team_id": 3, |
|
|
"away_team_id": 13, |
|
|
"home_score": 4, |
|
|
"away_score": 3, |
|
|
"attendance": 59387, |
|
|
"duration": 96, |
|
|
"referee": "Mike Dean", |
|
|
"venue": None, |
|
|
"home_manager": None, |
|
|
"away_manager": None, |
|
|
} |
|
|
OptaGameSchema.validate(pd.DataFrame.from_dict(games, orient="index")) |
|
|
|
|
|
|
|
|
def test_extract_teams(f9json_parser: F9JSONParser) -> None: |
|
|
teams = f9json_parser.extract_teams() |
|
|
assert len(teams) == 2 |
|
|
assert teams[3] == { |
|
|
"team_id": 3, |
|
|
"team_name": "Arsenal", |
|
|
} |
|
|
assert teams[13] == { |
|
|
"team_id": 13, |
|
|
"team_name": "Leicester City", |
|
|
} |
|
|
OptaTeamSchema.validate(pd.DataFrame.from_dict(teams, orient="index")) |
|
|
|
|
|
|
|
|
def test_extract_players(f9json_parser: F9JSONParser) -> None: |
|
|
players = f9json_parser.extract_players() |
|
|
assert len(players) == 27 |
|
|
assert players[(918893, 11334)] == { |
|
|
"game_id": 918893, |
|
|
"player_id": 11334, |
|
|
"player_name": "Petr Cech", |
|
|
"team_id": 3, |
|
|
"jersey_number": 33, |
|
|
"minutes_played": 96, |
|
|
"starting_position": "Goalkeeper", |
|
|
"is_starter": True, |
|
|
} |
|
|
OptaPlayerSchema.validate(pd.DataFrame.from_dict(players, orient="index")) |
|
|
|
