diff options
Diffstat (limited to 'tests/test_util_inspect.py')
| -rw-r--r-- | tests/test_util_inspect.py | 44 |
1 files changed, 24 insertions, 20 deletions
diff --git a/tests/test_util_inspect.py b/tests/test_util_inspect.py index f331acb23..8662fab12 100644 --- a/tests/test_util_inspect.py +++ b/tests/test_util_inspect.py @@ -157,21 +157,24 @@ def test_signature_annotations(): # Generic types with concrete parameters sig = inspect.signature(f1) - assert stringify_signature(sig) == '(x: List[int]) -> List[int]' + assert stringify_signature(sig) == '(x: typing.List[int]) -> typing.List[int]' # TypeVars and generic types with TypeVars sig = inspect.signature(f2) if sys.version_info < (3, 7): - assert stringify_signature(sig) == '(x: List[T], y: List[T_co], z: T) -> List[T_contra]' + assert stringify_signature(sig) == ('(x: typing.List[typing.T],' + ' y: typing.List[typing.T_co],' + ' z: typing.T' + ') -> typing.List[typing.T_contra]') else: - assert stringify_signature(sig) == ('(x: List[tests.typing_test_data.T],' - ' y: List[tests.typing_test_data.T_co],' + assert stringify_signature(sig) == ('(x: typing.List[tests.typing_test_data.T],' + ' y: typing.List[tests.typing_test_data.T_co],' ' z: tests.typing_test_data.T' - ') -> List[tests.typing_test_data.T_contra]') + ') -> typing.List[tests.typing_test_data.T_contra]') # Union types sig = inspect.signature(f3) - assert stringify_signature(sig) == '(x: Union[str, numbers.Integral]) -> None' + assert stringify_signature(sig) == '(x: typing.Union[str, numbers.Integral]) -> None' # Quoted annotations sig = inspect.signature(f4) @@ -187,18 +190,18 @@ def test_signature_annotations(): # Space around '=' for defaults sig = inspect.signature(f7) - assert stringify_signature(sig) == '(x: Optional[int] = None, y: dict = {}) -> None' + assert stringify_signature(sig) == '(x: typing.Optional[int] = None, y: dict = {}) -> None' # Callable types sig = inspect.signature(f8) - assert stringify_signature(sig) == '(x: Callable[[int, str], int]) -> None' + assert stringify_signature(sig) == '(x: typing.Callable[[int, str], int]) -> None' sig = inspect.signature(f9) - assert stringify_signature(sig) == '(x: Callable) -> None' + assert stringify_signature(sig) == '(x: typing.Callable) -> None' # Tuple types sig = inspect.signature(f10) - assert stringify_signature(sig) == '(x: Tuple[int, str], y: Tuple[int, ...]) -> None' + assert stringify_signature(sig) == '(x: typing.Tuple[int, str], y: typing.Tuple[int, ...]) -> None' # Instance annotations sig = inspect.signature(f11) @@ -206,24 +209,24 @@ def test_signature_annotations(): # tuple with more than two items sig = inspect.signature(f12) - assert stringify_signature(sig) == '() -> Tuple[int, str, int]' + assert stringify_signature(sig) == '() -> typing.Tuple[int, str, int]' # optional sig = inspect.signature(f13) - assert stringify_signature(sig) == '() -> Optional[str]' + assert stringify_signature(sig) == '() -> typing.Optional[str]' # optional union sig = inspect.signature(f20) - assert stringify_signature(sig) in ('() -> Optional[Union[int, str]]', - '() -> Optional[Union[str, int]]') + assert stringify_signature(sig) in ('() -> typing.Optional[typing.Union[int, str]]', + '() -> typing.Optional[typing.Union[str, int]]') # Any sig = inspect.signature(f14) - assert stringify_signature(sig) == '() -> Any' + assert stringify_signature(sig) == '() -> typing.Any' # ForwardRef sig = inspect.signature(f15) - assert stringify_signature(sig) == '(x: Unknown, y: int) -> Any' + assert stringify_signature(sig) == '(x: Unknown, y: int) -> typing.Any' # keyword only arguments (1) sig = inspect.signature(f16) @@ -234,7 +237,8 @@ def test_signature_annotations(): assert stringify_signature(sig) == '(*, arg3, arg4)' sig = inspect.signature(f18) - assert stringify_signature(sig) == '(self, arg1: Union[int, Tuple] = 10) -> List[Dict]' + assert stringify_signature(sig) == ('(self, arg1: typing.Union[int, typing.Tuple] = 10) -> ' + 'typing.List[typing.Dict]') # annotations for variadic and keyword parameters sig = inspect.signature(f19) @@ -246,10 +250,10 @@ def test_signature_annotations(): # type hints by string sig = inspect.signature(Node.children) - assert stringify_signature(sig) == '(self) -> List[tests.typing_test_data.Node]' + assert stringify_signature(sig) == '(self) -> typing.List[tests.typing_test_data.Node]' sig = inspect.signature(Node.__init__) - assert stringify_signature(sig) == '(self, parent: Optional[tests.typing_test_data.Node]) -> None' + assert stringify_signature(sig) == '(self, parent: typing.Optional[tests.typing_test_data.Node]) -> None' # show_annotation is False sig = inspect.signature(f7) @@ -257,7 +261,7 @@ def test_signature_annotations(): # show_return_annotation is False sig = inspect.signature(f7) - assert stringify_signature(sig, show_return_annotation=False) == '(x: Optional[int] = None, y: dict = {})' + assert stringify_signature(sig, show_return_annotation=False) == '(x: typing.Optional[int] = None, y: dict = {})' # unqualified_typehints is True sig = inspect.signature(f7) |