allow more errors when finding error line

samuelcolvin · Mar 21, 2023 · 2ba6953 · 2ba6953
1 parent c63e606
commit 2ba6953
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pytest_pretty/__init__.py b/pytest_pretty/__init__.py
@@ -77,12 +77,12 @@ def short_test_summary(self) -> None:
                 file, function_line, func = report.location
                 try:
                     repr_entries = report.longrepr.chain[-1][0].reprentries
+                    error_line = str(repr_entries[0].reprfileloc.lineno)
+                    error = repr_entries[-1].reprfileloc.message
                 except AttributeError:
                     error_line = ''
                     error = ''
-                else:
-                    error_line = str(repr_entries[0].reprfileloc.lineno)
-                    error = repr_entries[-1].reprfileloc.message
+
                 table.add_row(
                     escape(file),
                     escape(func),