From 4b297a289df06c43871b64c0c0e4b324a6749dba Mon Sep 17 00:00:00 2001 From: Firas Kafri <3097061+firaskafri@users.noreply.github.com> Date: Thu, 4 May 2023 22:05:05 +0300 Subject: [PATCH] Added tests --- graphene/types/tests/test_enum.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/graphene/types/tests/test_enum.py b/graphene/types/tests/test_enum.py index 9b3082df..b86d85d1 100644 --- a/graphene/types/tests/test_enum.py +++ b/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):