From 83d74e8f8b10df37ff1b4c7b29d634240f1a0e9d Mon Sep 17 00:00:00 2001
From: Koen Vossen <info@koenvossen.nl>
Date: Thu, 27 Jan 2022 10:52:51 +0100
Subject: [PATCH] Fix tests...

---
 kloppy/tests/test_opta.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/kloppy/tests/test_opta.py b/kloppy/tests/test_opta.py
index 9b37ff3c..f13f3d59 100644
--- 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"