diff --git a/graphene/tests/issues/test_1264.py b/graphene/tests/issues/test_1264.py deleted file mode 100644 index 48be0a8a..00000000 --- a/graphene/tests/issues/test_1264.py +++ /dev/null @@ -1,13 +0,0 @@ -from graphene.types.enum import Enum - - -def test_enum_iteration(): - class TestEnum(Enum): - FIRST = 1 - SECOND = 2 - - result = [] - expected_values = ["FIRST", "SECOND"] - for c in TestEnum: - result.append(c.name) - assert result == expected_values diff --git a/graphene/types/tests/test_enum.py b/graphene/types/tests/test_enum.py index ab1e093e..298cc233 100644 --- a/graphene/types/tests/test_enum.py +++ b/graphene/types/tests/test_enum.py @@ -543,3 +543,25 @@ def test_hashable_instance_creation_enum(): assert trilogy_map[Episode.NEWHOPE] == "better" assert trilogy_map[Episode.EMPIRE] == "best" assert trilogy_map[5] == "foo" + + +def test_enum_iteration(): + class TestEnum(Enum): + FIRST = 1 + SECOND = 2 + + result = [] + expected_values = ["FIRST", "SECOND"] + for c in TestEnum: + result.append(c.name) + assert result == expected_values + + +def test_iterable_instance_creation_enum(): + TestEnum = Enum("TestEnum", [("FIRST", 1), ("SECOND", 2)]) + + result = [] + expected_values = ["FIRST", "SECOND"] + for c in TestEnum: + result.append(c.name) + assert result == expected_values