Remove dependency on python3.11+ StrEnum from the test

smarie · Mar 19, 2024 · 26cc7aa · 26cc7aa
1 parent 690a461
commit 26cc7aa
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/test_issues.py b/tests/test_issues.py
@@ -283,12 +283,14 @@ def test_issue_91():
 
 
 def test_issue_98():
-    from enum import StrEnum
+    class A(str):
+        def __str__(self):
+            return 'custom str'
 
-    class A(StrEnum):
-        a = 'a'
+        def __repr__(self):
+            return 'custom repr'
 
-    def foo(a=A.a):
+    def foo(a=A()):
         pass
 
     @wraps(foo)