Fixed tests

Author: syrusakbary

graphql/error/tests/test_base.py

@@ -4,18 +4,19 @@
 from graphql.execution import execute
 from graphql.language.parser import parse
 from graphql.type import GraphQLField, GraphQLObjectType, GraphQLSchema, GraphQLString
-from graphql.execution.base import ResolveInfo
-from mypy_extensions import NoReturn
-from typing import Any
-from typing import Optional
+
+if False:
+    from graphql.execution.base import ResolveInfo
+    from typing import Any
+    from typing import Optional
 
 
 def test_raise():
     # type: () -> None
     ast = parse("query Example { a }")
 
     def resolver(context, *_):
-        # type: (Optional[Any], *ResolveInfo) -> NoReturn
+        # type: (Optional[Any], *ResolveInfo) -> None
         raise Exception("Failed")
 
     Type = GraphQLObjectType(
@@ -31,7 +32,7 @@ def test_reraise():
     ast = parse("query Example { a }")
 
     def resolver(context, *_):
-        # type: (Optional[Any], *ResolveInfo) -> NoReturn
+        # type: (Optional[Any], *ResolveInfo) -> None
         raise Exception("Failed")
 
     Type = GraphQLObjectType(

graphql/execution/tests/test_mutations.py

@@ -10,7 +10,6 @@
     GraphQLSchema,
     GraphQLString,
 )
-from mypy_extensions import NoReturn
 
 # from graphql.execution.executors.asyncio import AsyncioExecutor
 # from graphql.execution.executors.thread import ThreadExecutor
@@ -39,11 +38,11 @@ def promise_to_change_the_number(self, n):
         return self.immediately_change_the_number(n)
 
     def fail_to_change_the_number(self, n):
-        # type: (int) -> NoReturn
+        # type: (int) -> None
         raise Exception("Cannot change the number")
 
     def promise_and_fail_to_change_the_number(self, n):
-        # type: (int) -> NoReturn
+        # type: (int) -> None
         # TODO: async
         self.fail_to_change_the_number(n)
 

graphql/execution/tests/test_nonnull.py

@@ -12,13 +12,9 @@
 
 from .utils import rejected, resolved
 
-# from mypy_extensions import NoReturn
-# from promise.promise import Promise
-# from typing import Any
-# from typing import Optional
-# from typing import Dict
-# from typing import Tuple
-# from typing import Union
+if False:
+    from promise import Promise
+    from typing import Any, Optional, Dict, Tuple, Union
 
 sync_error = Exception("sync")
 non_null_sync_error = Exception("nonNullSync")
@@ -28,11 +24,11 @@
 
 class ThrowingData(object):
     def sync(self):
-        # type: () -> NoReturn
+        # type: () -> None
         raise sync_error
 
     def nonNullSync(self):
-        # type: () -> NoReturn
+        # type: () -> None
         raise non_null_sync_error
 
     def promise(self):

graphql/execution/tests/test_subscribe.py

@@ -15,17 +15,11 @@
     subscribe,
 )
 
-# from graphql.execution.base import ResolveInfo
-# from rx.core.anonymousobservable import AnonymousObservable
-# from rx.subjects.subject import Subject
-# from typing import Optional
-# from typing import Union
-# from mypy_extensions import NoReturn
-# from typing import Any
-# from typing import Callable
-# from graphql.type.schema import GraphQLSchema
-# from graphql.execution.base import ExecutionResult
-# from typing import Tuple
+if False:
+    from graphql.execution.base import ResolveInfo
+    from rx.core.anonymousobservable import AnonymousObservable
+    from typing import Optional, Union, Any, Callable, Tuple
+    from graphql.execution.base import ExecutionResult
 
 Email = namedtuple("Email", "from_,subject,message,unread")
 
@@ -277,7 +271,7 @@ def test_returns_an_error_if_subscribe_function_returns_error():
     exc = Exception("Throw!")
 
     def thrower(root, info):
-        # type: (Optional[Any], ResolveInfo) -> NoReturn
+        # type: (Optional[Any], ResolveInfo) -> None
         raise exc
 
     erroring_email_schema = email_schema_with_resolvers(thrower)