floraachy
/
pytest2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix overload of __getitem__ for Traceback

This commit is contained in:
Anthony Sottile 2021-06-19 21:41:00 -07:00
parent c4da6fff42
commit a1967e9594
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/_pytest/_code/code.py
View File

@ -48,6 +48,7 @@ from _pytest.pathlib import bestrelpath
if TYPE_CHECKING: if TYPE_CHECKING:
from typing_extensions import Literal from typing_extensions import Literal
from typing_extensions import SupportsIndex
from weakref import ReferenceType from weakref import ReferenceType
_TracebackStyle = Literal["long", "short", "line", "no", "native", "value", "auto"] _TracebackStyle = Literal["long", "short", "line", "no", "native", "value", "auto"]
@ -371,14 +372,16 @@ class Traceback(List[TracebackEntry]):
return self return self
@overload @overload
def __getitem__(self, key: int) -> TracebackEntry: def __getitem__(self, key: "SupportsIndex") -> TracebackEntry:
... ...
@overload @overload
def __getitem__(self, key: slice) -> "Traceback": def __getitem__(self, key: slice) -> "Traceback":
... ...
def __getitem__(self, key: Union[int, slice]) -> Union[TracebackEntry, "Traceback"]: def __getitem__(
self, key: Union["SupportsIndex", slice]
) -> Union[TracebackEntry, "Traceback"]:
if isinstance(key, slice): if isinstance(key, slice):
return self.__class__(super().__getitem__(key)) return self.__class__(super().__getitem__(key))
else: else: