18 lines
382 B
Python
18 lines
382 B
Python
import enum
|
|
|
|
from dopt_basics import enums
|
|
|
|
|
|
def test_enum_str_values_as_frzset():
|
|
class TestEnum(enum.StrEnum):
|
|
T1 = enum.auto()
|
|
T2 = enum.auto()
|
|
T3 = enum.auto()
|
|
|
|
target_vals = frozenset(("t1", "t2", "t3"))
|
|
extracted_vals = enums.enum_str_values_as_frzset(TestEnum)
|
|
|
|
diff = target_vals.difference(extracted_vals)
|
|
|
|
assert len(diff) == 0
|