fix types for unittest.TestLoader.loadTestsFrom* methods (#2042)

JelleZijlstra · ambv · commit 0f425db19779 · 2018-04-12T12:29:32.000-07:00
Fixes #2008.
diff --git 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]: ...
