Fix tests...

PySport · Jan 27, 2022 · 83d74e8 · 83d74e8
1 parent da6c2ef
commit 83d74e8
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/kloppy/tests/test_opta.py b/kloppy/tests/test_opta.py
@@ -14,6 +14,7 @@
     PassType,
     DatasetType,
     CardType,
+    FormationType,
 )
 
 from kloppy import opta
@@ -52,10 +53,14 @@ def test_correct_deserialization(self, f7_data: str, f24_data: str):
         )
         assert dataset.metadata.teams[0].name == "FC København"
         assert dataset.metadata.teams[0].ground == Ground.HOME
-        assert dataset.metadata.teams[0].starting_formation == "4-4-2"
+        assert dataset.metadata.teams[0].starting_formation == FormationType(
+            "4-4-2"
+        )
         assert dataset.metadata.teams[1].name == "FC Nordsjælland"
         assert dataset.metadata.teams[1].ground == Ground.AWAY
-        assert dataset.metadata.teams[1].starting_formation == "4-3-3"
+        assert dataset.metadata.teams[1].starting_formation == FormationType(
+            "4-3-3"
+        )
 
         player = dataset.metadata.teams[0].players[0]
         assert player.player_id == "111319"