Refactor enum implementation in test_enum.py

strawberry-graphql · Sep 23, 2024 · 73ce325 · 73ce325
1 parent 6c5ebc2
commit 73ce325
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/enums/test_enum.py b/tests/enums/test_enum.py
@@ -1,4 +1,4 @@
-from enum import Enum, IntEnum, StrEnum
+from enum import Enum, IntEnum
 
 import pytest
 
@@ -170,7 +170,7 @@ def foo(self, foo: Foo) -> Foo:
 
 
 def test_default_str_enum_implementation() -> None:
-    class Foo(StrEnum):
+    class Foo(str, Enum):
         BAR = "bar"
         BAZ = "baz"
 
@@ -188,7 +188,7 @@ def foo(self, foo: Foo) -> Foo:
 
 
 def test_default_int_enum_implementation() -> None:
-    class Foo(IntEnum):
+    class Foo(int, Enum):
         BAR = 1
         BAZ = 2