|
|
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) |
|
|
|
