mirror of
https://github.com/davidhalter/typeshed.git
synced 2025-12-08 13:04:46 +08:00
066c434410
* Improvements to stdlib/2/unittest.pyi * Adds a bunch of missing things. * Fixes the signatures of assertRaises and failUnlessRaises. * Corrects the name of a TestCase subclass from CallableTestCase to FunctionTestCase. * Makes defaultTestLoader an instance of TestLoader, as it should be, rather than an alias. Based on https://docs.python.org/2/library/unittest.html and https://github.com/python/cpython/tree/2.7/Lib/unittest. * Improvements to stdlib/3/unittest/__init__.pyi * The constructor arguments to TextTestResult are all required. * There is a module-level load_tests() function. * removeResult() returns a bool. Based on https://docs.python.org/3/library/unittest.html and https://github.com/python/cpython/tree/master/Lib/unittest.
249 lines
12 KiB
Python
249 lines
12 KiB
Python
# Stubs for unittest
|
|
|
|
# Based on http://docs.python.org/2.7/library/unittest.html
|
|
|
|
from mypy_extensions import NoReturn
|
|
from typing import (Any, Callable, Dict, FrozenSet, Iterable, Iterator,
|
|
List, Optional, overload, Pattern, Sequence, Set, Text,
|
|
TextIO, Tuple, Type, TypeVar, Union)
|
|
from abc import abstractmethod, ABCMeta
|
|
import types
|
|
|
|
_T = TypeVar('_T')
|
|
_FT = TypeVar('_FT')
|
|
|
|
_ExceptionType = Union[Type[BaseException], Tuple[Type[BaseException], ...]]
|
|
_Regexp = Union[Text, Pattern[Text]]
|
|
|
|
class Testable(metaclass=ABCMeta):
|
|
@abstractmethod
|
|
def run(self, result: 'TestResult') -> None: ...
|
|
@abstractmethod
|
|
def debug(self) -> None: ...
|
|
@abstractmethod
|
|
def countTestCases(self) -> int: ...
|
|
|
|
# TODO ABC for test runners?
|
|
|
|
class TestResult:
|
|
errors = ... # type: List[Tuple[Testable, str]]
|
|
failures = ... # type: List[Tuple[Testable, str]]
|
|
skipped = ... # type: List[Tuple[Testable, str]]
|
|
expectedFailures = ... # type: List[Tuple[Testable, str]]
|
|
unexpectedSuccesses = ... # type: List[Testable]
|
|
shouldStop = ... # type: bool
|
|
testsRun = ... # type: int
|
|
buffer = ... # type: bool
|
|
failfast = ... # type: bool
|
|
|
|
def wasSuccessful(self) -> bool: ...
|
|
def stop(self) -> None: ...
|
|
def startTest(self, test: Testable) -> None: ...
|
|
def stopTest(self, test: Testable) -> None: ...
|
|
def startTestRun(self) -> None: ...
|
|
def stopTestRun(self) -> None: ...
|
|
def addError(self, test: Testable,
|
|
err: Tuple[type, Any, Any]) -> None: ... # TODO
|
|
def addFailure(self, test: Testable,
|
|
err: Tuple[type, Any, Any]) -> None: ... # TODO
|
|
def addSuccess(self, test: Testable) -> None: ...
|
|
def addSkip(self, test: Testable, reason: str) -> None: ...
|
|
def addExpectedFailure(self, test: Testable, err: str) -> None: ...
|
|
def addUnexpectedSuccess(self, test: Testable) -> None: ...
|
|
|
|
class _AssertRaisesBaseContext:
|
|
expected = ... # type: Any
|
|
failureException = ... # type: Type[BaseException]
|
|
obj_name = ... # type: str
|
|
expected_regex = ... # type: Pattern[str]
|
|
|
|
class _AssertRaisesContext(_AssertRaisesBaseContext):
|
|
exception = ... # type: Any # TODO precise type
|
|
def __enter__(self) -> _AssertRaisesContext: ...
|
|
def __exit__(self, exc_type, exc_value, tb) -> bool: ...
|
|
|
|
class TestCase(Testable):
|
|
failureException = ... # type: Type[BaseException]
|
|
longMessage = ... # type: bool
|
|
maxDiff = ... # type: Optional[int]
|
|
def __init__(self, methodName: str = ...) -> None: ...
|
|
def setUp(self) -> None: ...
|
|
def tearDown(self) -> None: ...
|
|
@classmethod
|
|
def setUpClass(cls) -> None: ...
|
|
@classmethod
|
|
def tearDownClass(cls) -> None: ...
|
|
def run(self, result: TestResult = ...) -> None: ...
|
|
def debug(self) -> None: ...
|
|
def assert_(self, expr: Any, msg: object = ...) -> None: ...
|
|
def failUnless(self, expr: Any, msg: object = ...) -> None: ...
|
|
def assertTrue(self, expr: Any, msg: object = ...) -> None: ...
|
|
def assertEqual(self, first: Any, second: Any,
|
|
msg: object = ...) -> None: ...
|
|
def assertEquals(self, first: Any, second: Any,
|
|
msg: object = ...) -> None: ...
|
|
def failUnlessEqual(self, first: Any, second: Any,
|
|
msg: object = ...) -> None: ...
|
|
def assertNotEqual(self, first: Any, second: Any,
|
|
msg: object = ...) -> None: ...
|
|
def assertNotEquals(self, first: Any, second: Any,
|
|
msg: object = ...) -> None: ...
|
|
def failIfEqual(self, first: Any, second: Any,
|
|
msg: object = ...) -> None: ...
|
|
def assertAlmostEqual(self, first: float, second: float, places: int = ...,
|
|
msg: object = ...,
|
|
delta: float = ...) -> None: ...
|
|
def assertAlmostEquals(self, first: float, second: float, places: int = ...,
|
|
msg: object = ...,
|
|
delta: float = ...) -> None: ...
|
|
def failUnlessAlmostEqual(self, first: float, second: float, places: int = ...,
|
|
msg: object = ...) -> None: ...
|
|
def assertNotAlmostEqual(self, first: float, second: float, places: int = ...,
|
|
msg: object = ...,
|
|
delta: float = ...) -> None: ...
|
|
def assertNotAlmostEquals(self, first: float, second: float, places: int = ...,
|
|
msg: object = ...,
|
|
delta: float = ...) -> None: ...
|
|
def failIfAlmostEqual(self, first: float, second: float, places: int = ...,
|
|
msg: object = ...,
|
|
delta: float = ...) -> None: ...
|
|
def assertGreater(self, first: Any, second: Any,
|
|
msg: object = ...) -> None: ...
|
|
def assertGreaterEqual(self, first: Any, second: Any,
|
|
msg: object = ...) -> None: ...
|
|
def assertMultiLineEqual(self, first: str, second: str,
|
|
msg: object = ...) -> None: ...
|
|
def assertSequenceEqual(self, first: Sequence[Any], second: Sequence[Any],
|
|
msg: object = ..., seq_type: type = ...) -> None: ...
|
|
def assertListEqual(self, first: List[Any], second: List[Any],
|
|
msg: object = ...) -> None: ...
|
|
def assertTupleEqual(self, first: Tuple[Any, ...], second: Tuple[Any, ...],
|
|
msg: object = ...) -> None: ...
|
|
def assertSetEqual(self, first: Union[Set[Any], FrozenSet[Any]],
|
|
second: Union[Set[Any], FrozenSet[Any]], msg: object = ...) -> None: ...
|
|
def assertDictEqual(self, first: Dict[Any, Any], second: Dict[Any, Any],
|
|
msg: object = ...) -> None: ...
|
|
def assertLess(self, first: Any, second: Any,
|
|
msg: object = ...) -> None: ...
|
|
def assertLessEqual(self, first: Any, second: Any,
|
|
msg: object = ...) -> None: ...
|
|
@overload
|
|
def assertRaises(self, exception: _ExceptionType, callable: Callable[..., Any], *args: Any, **kwargs: Any) -> None: ...
|
|
@overload
|
|
def assertRaises(self, exception: _ExceptionType) -> _AssertRaisesContext: ...
|
|
@overload
|
|
def assertRaisesRegexp(self, exception: _ExceptionType, regexp: _Regexp, callable: Callable[..., Any], *args: Any, **kwargs: Any) -> None: ...
|
|
@overload
|
|
def assertRaisesRegexp(self, exception: _ExceptionType, regexp: _Regexp) -> _AssertRaisesContext: ...
|
|
def assertRegexpMatches(self, text: Text, regexp: _Regexp, msg: object = ...) -> None: ...
|
|
def assertNotRegexpMatches(self, text: Text, regexp: _Regexp, msg: object = ...) -> None: ...
|
|
def assertItemsEqual(self, first: Iterable[Any], second: Iterable[Any], msg: object = ...) -> None: ...
|
|
def assertDictContainsSubset(self, expected: Dict[Any, Any], actual: Dict[Any, Any], msg: object = ...) -> None: ...
|
|
def addTypeEqualityFunc(self, typeobj: type, function: Callable[..., None]) -> None: ...
|
|
@overload
|
|
def failUnlessRaises(self, exception: _ExceptionType, callable: Callable[..., Any], *args: Any, **kwargs: Any) -> None: ...
|
|
@overload
|
|
def failUnlessRaises(self, exception: _ExceptionType) -> _AssertRaisesContext: ...
|
|
def failIf(self, expr: Any, msg: object = ...) -> None: ...
|
|
def assertFalse(self, expr: Any, msg: object = ...) -> None: ...
|
|
def assertIs(self, first: object, second: object,
|
|
msg: object = ...) -> None: ...
|
|
def assertIsNot(self, first: object, second: object,
|
|
msg: object = ...) -> None: ...
|
|
def assertIsNone(self, expr: Any, msg: object = ...) -> None: ...
|
|
def assertIsNotNone(self, expr: Any, msg: object = ...) -> None: ...
|
|
def assertIn(self, first: _T, second: Iterable[_T],
|
|
msg: object = ...) -> None: ...
|
|
def assertNotIn(self, first: _T, second: Iterable[_T],
|
|
msg: object = ...) -> None: ...
|
|
def assertIsInstance(self, obj: Any, cls: Union[type, Tuple[type, ...]],
|
|
msg: object = ...) -> None: ...
|
|
def assertNotIsInstance(self, obj: Any, cls: Union[type, Tuple[type, ...]],
|
|
msg: object = ...) -> None: ...
|
|
def fail(self, msg: object = ...) -> NoReturn: ...
|
|
def countTestCases(self) -> int: ...
|
|
def defaultTestResult(self) -> TestResult: ...
|
|
def id(self) -> str: ...
|
|
def shortDescription(self) -> str: ... # May return None
|
|
def addCleanup(function: Any, *args: Any, **kwargs: Any) -> None: ...
|
|
def doCleanups(self) -> bool: ...
|
|
def skipTest(self, reason: Any) -> None: ...
|
|
|
|
class FunctionTestCase(Testable):
|
|
def __init__(self, testFunc: Callable[[], None],
|
|
setUp: Optional[Callable[[], None]] = ...,
|
|
tearDown: Optional[Callable[[], None]] = ...,
|
|
description: Optional[str] = ...) -> None: ...
|
|
def run(self, result: TestResult) -> None: ...
|
|
def debug(self) -> None: ...
|
|
def countTestCases(self) -> int: ...
|
|
|
|
class TestSuite(Testable):
|
|
def __init__(self, tests: Iterable[Testable] = ...) -> None: ...
|
|
def addTest(self, test: Testable) -> None: ...
|
|
def addTests(self, tests: Iterable[Testable]) -> None: ...
|
|
def run(self, result: TestResult) -> None: ...
|
|
def debug(self) -> None: ...
|
|
def countTestCases(self) -> int: ...
|
|
def __iter__(self) -> Iterator[Testable]: ...
|
|
|
|
class TestLoader:
|
|
testMethodPrefix = ... # type: str
|
|
sortTestMethodsUsing = ... # type: Optional[Callable[[str, str], int]]
|
|
suiteClass = ... # type: Callable[[List[TestCase]], TestSuite]
|
|
def loadTestsFromTestCase(self,
|
|
testCaseClass: Type[TestCase]) -> TestSuite: ...
|
|
def loadTestsFromModule(self, module: str = ...,
|
|
use_load_tests: bool = ...) -> TestSuite: ...
|
|
def loadTestsFromName(self, name: str = ...,
|
|
module: Optional[str] = ...) -> TestSuite: ...
|
|
def loadTestsFromNames(self, names: List[str] = ...,
|
|
module: Optional[str] = ...) -> TestSuite: ...
|
|
def discover(self, start_dir: str, pattern: str = ...,
|
|
top_level_dir: Optional[str] = ...) -> TestSuite: ...
|
|
def getTestCaseNames(self, testCaseClass: Type[TestCase] = ...) -> List[str]: ...
|
|
|
|
defaultTestLoader = ... # type: TestLoader
|
|
|
|
class TextTestResult(TestResult):
|
|
def __init__(self, stream: TextIO, descriptions: bool, verbosity: int) -> None: ...
|
|
|
|
class TextTestRunner:
|
|
def __init__(self, stream: Optional[TextIO] = ..., descriptions: bool = ...,
|
|
verbosity: int = ..., failfast: bool = ..., buffer: bool = ...,
|
|
resultclass: Optional[Type[TestResult]] = ...) -> None: ...
|
|
def _makeResult(self) -> TestResult: ...
|
|
|
|
class SkipTest(Exception):
|
|
...
|
|
|
|
# TODO precise types
|
|
def skipUnless(condition: Any, reason: Union[str, unicode]) -> Any: ...
|
|
def skipIf(condition: Any, reason: Union[str, unicode]) -> Any: ...
|
|
def expectedFailure(func: _FT) -> _FT: ...
|
|
def skip(reason: Union[str, unicode]) -> Any: ...
|
|
|
|
# not really documented
|
|
class TestProgram:
|
|
result = ... # type: TestResult
|
|
|
|
def main(module: str = ..., defaultTest: Optional[str] = ...,
|
|
argv: Optional[Sequence[str]] = ...,
|
|
testRunner: Union[Type[TextTestRunner], TextTestRunner, None] = ...,
|
|
testLoader: TestLoader = ..., exit: bool = ..., verbosity: int = ...,
|
|
failfast: Optional[bool] = ..., catchbreak: Optional[bool] = ...,
|
|
buffer: Optional[bool] = ...) -> TestProgram: ...
|
|
|
|
def load_tests(loader: TestLoader, tests: TestSuite, pattern: Optional[Text]) -> TestSuite: ...
|
|
|
|
def installHandler() -> None: ...
|
|
def registerResult(result: TestResult) -> None: ...
|
|
def removeResult(result: TestResult) -> bool: ...
|
|
@overload
|
|
def removeHandler() -> None: ...
|
|
@overload
|
|
def removeHandler(function: Callable[..., Any]) -> Callable[..., Any]: ...
|
|
|
|
# private but occasionally used
|
|
util = ... # type: types.ModuleType
|