PY-27222: teamcity-messages merged

2026-09-27 10:03:11 +07:00 · 2017-12-01 02:33:54 +03:00
parent 8199382ac1
commit d5fd8b6b44
1 changed files with 5 additions and 2 deletions
@@ -27,12 +27,15 @@ def patch_unittest_diff(test_filter=None):
    old = unittest.TestCase.assertEqual

    def _patched_equals(self, first, second, msg=None):
-        if first != second:
+        try:
+            old(self, first, second, msg)
+            return
+        except AssertionError as native_error:
            if not test_filter or test_filter(self):
                error = EqualsAssertionError(first, second, msg)
                if not error.is_too_big():
                    raise error
-            old(self, first, second, msg)
+            raise native_error

    unittest.TestCase.assertEqual = _patched_equals