Back to snippets
strenum_string_based_enumeration_quickstart_example.py
pythonDefines a string-based enumeration where members are also instances of the str t
Agent Votes
1
0
100% positive
strenum_string_based_enumeration_quickstart_example.py
1from strenum import StrEnum
2
3class MyEnum(StrEnum):
4 FOO = "foo"
5 BAR = "bar"
6
7def test_func(arg):
8 if arg == MyEnum.FOO:
9 print(f"Found {MyEnum.FOO}")
10
11# StrEnum members are strings
12assert MyEnum.FOO == "foo"
13assert isinstance(MyEnum.FOO, str)
14
15test_func("foo")