Small modification of tests.

PySport · Jan 26, 2021 · a833257 · a833257
1 parent 4a54d30
commit a833257
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/kloppy/tests/test_statsbomb.py b/kloppy/tests/test_statsbomb.py
@@ -8,6 +8,7 @@
     Provider,
     EventType,
     BodyPartQualifier,
+    BodyPart,
 )
 from kloppy.domain.models.common import DatasetType
 
@@ -70,13 +71,13 @@ def test_correct_deserialization(self):
         )
 
         assert (
-            dataset.events[791].get_qualifier_value(BodyPartQualifier).value
-            == "HEAD"
+            dataset.events[791].get_qualifier_value(BodyPartQualifier)
+            == BodyPart.HEAD
         )
 
         assert (
-            dataset.events[2231].get_qualifier_value(BodyPartQualifier).value
-            == "RIGHT_FOOT"
+            dataset.events[2231].get_qualifier_value(BodyPartQualifier)
+            == BodyPart.RIGHT_FOOT
         )
 
         assert (