Skip to content

Commit f0448c3

Browse files
author
Sylvain MARIE
committed
Improved type hint for @parametrize, see #245
1 parent 3c8b16a commit f0448c3

File tree

1 file changed

+17
-13
lines changed

1 file changed

+17
-13
lines changed

src/pytest_cases/fixture_parametrize_plus.py

Lines changed: 17 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
from collections import Iterable
1818

1919
try:
20-
from typing import Union, Callable, List, Any, Sequence, Optional, Type # noqa
20+
from typing import Union, Callable, List, Any, Sequence, Optional, Type, Tuple # noqa
2121
from types import ModuleType # noqa
2222
except ImportError:
2323
pass
@@ -613,7 +613,7 @@ def nostyle(cls,
613613
_IDGEN = object()
614614

615615

616-
def parametrize(argnames=None, # type: Union[str, List[str]]
616+
def parametrize(argnames=None, # type: Union[str, Tuple[str], List[str]]
617617
argvalues=None, # type: Iterable[Any]
618618
indirect=False, # type: bool
619619
ids=None, # type: Union[Callable, Iterable[str]]
@@ -722,16 +722,16 @@ def __repr__(self):
722722
% (self.idgen, self.params, self.caught.__class__, self.caught)
723723

724724

725-
def _parametrize_plus(argnames=None,
726-
argvalues=None,
727-
indirect=False, # type: bool
728-
ids=None, # type: Union[Callable, Iterable[str]]
729-
idstyle=None, # type: Optional[Union[str, Callable]]
730-
idgen=_IDGEN, # type: Union[str, Callable]
731-
auto_refs=True, # type: bool
732-
scope=None, # type: str
733-
hook=None, # type: Callable[[Callable], Callable]
734-
debug=False, # type: bool
725+
def _parametrize_plus(argnames=None, # type: Union[str, Tuple[str], List[str]]
726+
argvalues=None, # type: Iterable[Any]
727+
indirect=False, # type: bool
728+
ids=None, # type: Union[Callable, Iterable[str]]
729+
idstyle=None, # type: Optional[Union[str, Callable]]
730+
idgen=_IDGEN, # type: Union[str, Callable]
731+
auto_refs=True, # type: bool
732+
scope=None, # type: str
733+
hook=None, # type: Callable[[Callable], Callable]
734+
debug=False, # type: bool
735735
**args):
736736
"""
737737
@@ -1096,7 +1096,11 @@ def wrapped_test_func(*args, **kwargs): # noqa
10961096
return parametrize_plus_decorate, True
10971097

10981098

1099-
def _get_argnames_argvalues(argnames=None, argvalues=None, **args):
1099+
def _get_argnames_argvalues(
1100+
argnames=None, # type: Union[str, Tuple[str], List[str]]
1101+
argvalues=None, # type: Iterable[Any]
1102+
**args
1103+
):
11001104
"""
11011105
11021106
:param argnames:

0 commit comments

Comments
 (0)