diff --git a/neotest_python/unittest.py b/neotest_python/unittest.py index 6c58aeb..d5d45ec 100644 --- a/neotest_python/unittest.py +++ b/neotest_python/unittest.py @@ -72,11 +72,10 @@ class UnittestNeotestAdapter(NeotestAdapter): if case_id in errs: trace = errs[case_id][2] summary = traceback.extract_tb(trace) - error_line = next( - frame.lineno - 1 - for frame in reversed(summary) - if frame.filename == case_file - ) + for frame in reversed(summary): + if frame.filename == case_file: + error_line = frame.lineno - 1 + break results[case_id] = { "status": NeotestResultStatus.FAILED, "errors": [{"message": message, "line": error_line}],