Skip to content

Commit 683a71f

Browse files
committed
fix: StrEnum not available before 3.11
1 parent 10a89a9 commit 683a71f

File tree

1 file changed

+15
-9
lines changed

1 file changed

+15
-9
lines changed

tests/test_enum.py

Lines changed: 15 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from enum import Enum, StrEnum
1+
from enum import Enum
22
from unittest import TestCase, skipIf
33

44
from transitions.core import Machine
@@ -398,17 +398,23 @@ class States(enum.Enum):
398398
self.assertEqual(ref_state, m.state)
399399

400400

401-
class TestNestedStateStrEnums(TestNestedStateEnums):
401+
try:
402+
from enum import StrEnum
402403

403-
def setUp(self):
404-
super().setUp()
404+
class TestNestedStateStrEnums(TestNestedStateEnums):
405+
406+
def setUp(self):
407+
super().setUp()
405408

406-
class States(StrEnum):
407-
RED = "red"
408-
YELLOW = "yellow"
409-
GREEN = "green"
409+
class States(StrEnum):
410+
RED = "red"
411+
YELLOW = "yellow"
412+
GREEN = "green"
410413

411-
self.States = States
414+
self.States = States
415+
416+
except ImportError:
417+
pass # Python < 3.11 does not have StrEnum
412418

413419

414420
@skipIf(enum is None or (pgv is None and gv is None), "enum and (py)graphviz are not available")

0 commit comments

Comments
 (0)