Fix test_description

graphql-python · Sep 3, 2024 · 666cf03 · 666cf03
1 parent 9d0b271
commit 666cf03
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/tests/pyutils/test_description.py b/tests/pyutils/test_description.py
@@ -1,27 +1,26 @@
 from contextlib import contextmanager
 from typing import cast
 
-from pytest import raises
-
 from graphql import graphql_sync
+from graphql.pyutils import (
+    Description,
+    is_description,
+    register_description,
+    unregister_description,
+)
 from graphql.type import (
     GraphQLArgument,
     GraphQLDirective,
     GraphQLEnumValue,
     GraphQLField,
     GraphQLInputField,
-    GraphQLObjectType,
     GraphQLNamedType,
+    GraphQLObjectType,
     GraphQLSchema,
     GraphQLString,
 )
-from graphql.pyutils import (
-    Description,
-    is_description,
-    register_description,
-    unregister_description,
-)
 from graphql.utilities import get_introspection_query, print_schema
+from pytest import raises
 
 from ..utils import dedent
 
@@ -43,7 +42,7 @@ def registered(base: type):
     try:
         yield None
     finally:
-        unregister_description(LazyString)
+        unregister_description(base)
 
 
 def describe_description():