Skip to content

Commit

Permalink
Adding test which checks orientation of Wyscout generated EventDataset
Browse files Browse the repository at this point in the history
DriesDeprest committed Jan 12, 2024
1 parent 8f2b20c commit 93d508a
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions kloppy/tests/test_wyscout.py
Original file line number Diff line number Diff line change
@@ -16,6 +16,7 @@
GoalkeeperActionType,
CardQualifier,
CardType,
Orientation,
)

from kloppy import wyscout
@@ -52,6 +53,9 @@ def dataset(self, event_v2_data) -> EventDataset:
data_version="V2",
)
assert dataset.dataset_type == DatasetType.EVENT
assert (
dataset.metadata.orientation == Orientation.ACTION_EXECUTING_TEAM
)
return dataset

def test_shot_event(self, dataset: EventDataset):
@@ -130,6 +134,9 @@ def dataset(self, event_v3_data: Path) -> EventDataset:
data_version="V3",
)
assert dataset.dataset_type == DatasetType.EVENT
assert (
dataset.metadata.orientation == Orientation.ACTION_EXECUTING_TEAM
)
return dataset

def test_coordinates(self, dataset: EventDataset):

0 comments on commit 93d508a

Please sign in to comment.