| import os |
| from datetime import datetime |
|
|
| import pandas as pd |
| from pytest import fixture |
| from socceraction.data.opta import OptaEventSchema, OptaGameSchema |
| from socceraction.data.opta.parsers import F24XMLParser |
|
|
|
|
| @fixture() |
| def f24xml_parser() -> F24XMLParser: |
| path = os.path.join( |
| os.path.dirname(__file__), |
| os.pardir, |
| os.pardir, |
| os.pardir, |
| "datasets", |
| "opta", |
| "f24-23-2018-1009316-eventdetails.xml", |
| ) |
| return F24XMLParser(str(path)) |
|
|
|
|
| def test_extract_games(f24xml_parser: F24XMLParser) -> None: |
| games = f24xml_parser.extract_games() |
| assert len(games) == 1 |
| assert games[1009316] == { |
| "game_id": 1009316, |
| "season_id": 2018, |
| "competition_id": 23, |
| "game_day": 1, |
| "game_date": datetime(2018, 8, 20, 21, 0), |
| "home_team_id": 174, |
| "away_team_id": 957, |
| "home_score": 2, |
| "away_score": 1, |
| } |
| OptaGameSchema.validate(pd.DataFrame.from_dict(games, orient="index")) |
|
|
|
|
| def test_extract_events(f24xml_parser: F24XMLParser) -> None: |
| events = f24xml_parser.extract_events() |
| assert len(events) == 1665 |
| assert events[(1009316, 2097423126)] == { |
| "game_id": 1009316, |
| "event_id": 2097423126, |
| "period_id": 2, |
| "team_id": 174, |
| "player_id": 197319, |
| "type_id": 1, |
| "timestamp": datetime(2018, 8, 20, 22, 51, 28, 259000), |
| "minute": 94, |
| "second": 50, |
| "outcome": False, |
| "start_x": 46.4, |
| "start_y": 37.1, |
| "end_x": 74.4, |
| "end_y": 8.9, |
| "qualifiers": { |
| 1: None, |
| 213: "5.7", |
| 212: "35.1", |
| 152: None, |
| 5: None, |
| 155: None, |
| 56: "Right", |
| 140: "74.4", |
| 141: "8.9", |
| }, |
| "assist": False, |
| "keypass": False, |
| } |
| df = pd.DataFrame.from_dict(events, orient="index") |
| df["type_name"] = "Added later" |
| OptaEventSchema.validate(df) |
|
|
