From be502f932906e19a418285b2e57b2486435938e4 Mon Sep 17 00:00:00 2001 From: Cheukting Date: Thu, 23 Jun 2022 15:48:21 +0800 Subject: [PATCH] Fix MyPy --- src/_pytest/_code/code.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/src/_pytest/_code/code.py b/src/_pytest/_code/code.py index 3fa0b637b..9f04c0ffe 100644 --- a/src/_pytest/_code/code.py +++ b/src/_pytest/_code/code.py @@ -56,9 +56,9 @@ if TYPE_CHECKING: _TracebackStyle = Literal["long", "short", "line", "no", "native", "value", "auto"] -ExceptionGroupTypes: tuple = () +ExceptionGroupTypes: tuple = () # type: ignore try: - ExceptionGroupTypes += (ExceptionGroup,) + ExceptionGroupTypes += (ExceptionGroup,) # type: ignore except NameError: pass # Is missing for `python<3.10` try: @@ -68,7 +68,6 @@ try: except ModuleNotFoundError: pass # No backport is installed - class Code: """Wrapper around Python code objects.""" @@ -936,15 +935,15 @@ class FormattedExcinfo: while e is not None and id(e) not in seen: seen.add(id(e)) if isinstance(e, ExceptionGroupTypes): - reprtraceback = ReprTracebackNative( + reprtraceback: Union[ReprTracebackNative, ReprTraceback] = ReprTracebackNative( traceback.format_exception( - type(e), e, excinfo.traceback[0]._rawentry + type(excinfo.value), excinfo.value, excinfo.traceback[0]._rawentry ) ) - reprcrash = None + reprcrash: Optional[ReprFileLocation] = None elif excinfo_: reprtraceback = self.repr_traceback(excinfo_) - reprcrash: Optional[ReprFileLocation] = ( + reprcrash = ( excinfo_._getreprcrash() if self.style != "value" else None ) else: