File size: 2,015 Bytes
d6ea71e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
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)
|