Added tests

Author: firaskafri

graphene/types/tests/test_enum.py

@@ -64,6 +64,19 @@ def test_enum_from_builtin_enum():
     assert RGB.GREEN
     assert RGB.BLUE
 
+def test_enum_custom_description_in_constructor():
+    description = "An enumeration, but with a custom description"
+    RGB = Enum(
+        "RGB",
+        "RED,GREEN,BLUE",
+        description=description,
+    )
+    assert RGB._meta.description == description
+
+
+def test_enum_from_python3_enum_uses_default_builtin_doc():
+    RGB = Enum("RGB", "RED,GREEN,BLUE")
+    assert RGB._meta.description == "An enumeration."
 
 def test_enum_from_builtin_enum_accepts_lambda_description():
     def custom_description(value):