Update typing for values in EnumType.__init__ (mirumee#1040)

Co-authored-by: Mark Edwards <me@g18e.com>

ariadne/enums.py

@@ -7,6 +7,7 @@
     Mapping,
     Optional,
     Tuple,
+    Type,
     TypeVar,
     Union,
     cast,
@@ -87,7 +88,9 @@ class EnumType(SchemaBindable):
     """
 
     def __init__(
-        self, name: str, values: Union[Dict[str, Any], enum.Enum, enum.IntEnum]
+        self,
+        name: str,
+        values: Union[Dict[str, Any], Type[enum.Enum], Type[enum.IntEnum]],
     ) -> None:
         """Initializes the `EnumType` with `name` and `values` mapping.
 

tests_mypy/enums.py

@@ -0,0 +1,18 @@
+from enum import Enum, IntEnum
+
+from ariadne import EnumType
+
+
+class TestEnum(Enum):
+    FOO = "foo"
+    BAR = "bar"
+
+
+class TestIntEnum(IntEnum):
+    FOO = 0
+    BAR = 1
+
+
+test_dict_enum_type = EnumType("TestDictEnum", {"FOO": "foo", "BAR": "bar"})
+test_enum_type = EnumType("TestEnum", TestEnum)
+test_int_enum_type = EnumType("TestIntEnumType", TestIntEnum)