From 0f425db197791a04d85bfa6b4b269e70421b27e5 Mon Sep 17 00:00:00 2001
From: Jelle Zijlstra <jelle.zijlstra@gmail.com>
Date: Thu, 12 Apr 2018 12:29:32 -0700
Subject: [PATCH] fix types for unittest.TestLoader.loadTestsFrom* methods
 (#2042)

Fixes #2008.
---
 stdlib/2/unittest.pyi | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/stdlib/2/unittest.pyi b/stdlib/2/unittest.pyi
index f79ec5e86..8e8665034 100644
--- a/stdlib/2/unittest.pyi
+++ b/stdlib/2/unittest.pyi
@@ -192,12 +192,12 @@ class TestLoader:
     suiteClass = ...  # type: Callable[[List[TestCase]], TestSuite]
     def loadTestsFromTestCase(self,
                               testCaseClass: Type[TestCase]) -> TestSuite: ...
-    def loadTestsFromModule(self, module: str = ...,
+    def loadTestsFromModule(self, module: types.ModuleType = ...,
                             use_load_tests: bool = ...) -> TestSuite: ...
     def loadTestsFromName(self, name: str = ...,
-                          module: Optional[str] = ...) -> TestSuite: ...
+                          module: Optional[types.ModuleType] = ...) -> TestSuite: ...
     def loadTestsFromNames(self, names: List[str] = ...,
-                          module: Optional[str] = ...) -> TestSuite: ...
+                          module: Optional[types.ModuleType] = ...) -> TestSuite: ...
     def discover(self, start_dir: str, pattern: str = ...,
                  top_level_dir: Optional[str] = ...) -> TestSuite: ...
     def getTestCaseNames(self, testCaseClass: Type[TestCase] = ...) -> List[str]: ...