From 5efd77728dd5a1c6a2fb5070b981ad4d028c3fe1 Mon Sep 17 00:00:00 2001 From: Paul Harrison Date: Fri, 18 Nov 2022 12:59:18 +0000 Subject: feat: Implement auto-naming string enum --- tests/utils/__init__.py | 0 tests/utils/test_enum.py | 16 ++++++++++++++++ 2 files changed, 16 insertions(+) create mode 100644 tests/utils/__init__.py create mode 100644 tests/utils/test_enum.py (limited to 'tests') diff --git a/tests/utils/__init__.py b/tests/utils/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/utils/test_enum.py b/tests/utils/test_enum.py new file mode 100644 index 0000000..9a84ee1 --- /dev/null +++ b/tests/utils/test_enum.py @@ -0,0 +1,16 @@ +from enum import auto + +from poker.utils.enum import AutoName + + +def test_auto_name() -> None: + class Example(AutoName): + """Example Enum.""" + + foo = auto() + BAR = auto() + WeIrD = auto() + + assert Example.foo.value == "foo" + assert Example.BAR.value == "bar" + assert Example.WeIrD.value == "weird" -- cgit v1.2.3