From a8332571e22f4e7ace0d2571c56254803dbf9d57 Mon Sep 17 00:00:00 2001 From: BenjaminLarrousse Date: Tue, 26 Jan 2021 12:22:33 +0100 Subject: [PATCH] Small modification of tests. --- kloppy/tests/test_statsbomb.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/kloppy/tests/test_statsbomb.py b/kloppy/tests/test_statsbomb.py index abec65d6..36b1ddc5 100644 --- 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 (