From 8b9e4c378a5421dce750c23302e22a443676ad81 Mon Sep 17 00:00:00 2001 From: Jelle Zijlstra Date: Mon, 10 Sep 2018 11:04:05 -0700 Subject: [PATCH] fix module argument to unittest.main (#2403) --- stdlib/2/unittest.pyi | 2 +- stdlib/3/unittest/__init__.pyi | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/stdlib/2/unittest.pyi b/stdlib/2/unittest.pyi index 43f08a45b..a5cd129b0 100644 --- a/stdlib/2/unittest.pyi +++ b/stdlib/2/unittest.pyi @@ -226,7 +226,7 @@ def skip(reason: Union[str, unicode]) -> Any: ... class TestProgram: result = ... # type: TestResult -def main(module: str = ..., defaultTest: Optional[str] = ..., +def main(module: Union[None, Text, types.ModuleType] = ..., defaultTest: Optional[str] = ..., argv: Optional[Sequence[str]] = ..., testRunner: Union[Type[TextTestRunner], TextTestRunner, None] = ..., testLoader: TestLoader = ..., exit: bool = ..., verbosity: int = ..., diff --git a/stdlib/3/unittest/__init__.pyi b/stdlib/3/unittest/__init__.pyi index 5b7b37352..3337d00ba 100644 --- a/stdlib/3/unittest/__init__.pyi +++ b/stdlib/3/unittest/__init__.pyi @@ -307,7 +307,7 @@ class TextTestRunner(TestRunner): class TestProgram: result = ... # type: TestResult -def main(module: str = ..., +def main(module: Union[None, str, ModuleType] = ..., defaultTest: Union[str, Iterable[str], None] = ..., argv: Optional[List[str]] = ..., testRunner: Union[Type[TestRunner], TestRunner, None] = ...,