diff --git a/mypy.ini b/mypy.ini index d8ec951ff0..b708fa621c 100644 --- a/mypy.ini +++ b/mypy.ini @@ -36,9 +36,6 @@ ignore_missing_imports = True [mypy-lister,.lister] ignore_missing_imports = True -[mypy-zerver.lib.test_case] -ignore_missing_imports = True - # Zulip code from elsewhere diff --git a/zerver/lib/test_helpers.py b/zerver/lib/test_helpers.py index 4da877b352..db81044dec 100644 --- a/zerver/lib/test_helpers.py +++ b/zerver/lib/test_helpers.py @@ -46,9 +46,7 @@ from zerver.models import ( ) from zerver.lib.request import JsonableError - -if False: - from zerver.lib.test_case import ZulipTestCase +from zerver.lib.test_classes import ZulipTestCase import collections import base64 @@ -291,7 +289,7 @@ def instrument_url(f: UrlFuncT) -> UrlFuncT: if not INSTRUMENTING: # nocoverage -- option is always enabled; should we remove? return f else: - def wrapper(self: 'ZulipTestCase', url: str, info: Dict[str, Any]={}, + def wrapper(self: ZulipTestCase, url: str, info: Dict[str, Any]={}, **kwargs: Any) -> HttpResponse: start = time.time() result = f(self, url, info, **kwargs)