Extend test

ShaharNaveh · ShaharNaveh · commit 2bf0b6ae23cb · 2026-06-07T14:34:11.000+03:00
diff --git a/scripts/update_lib/tests/test_patch_spec.py b/scripts/update_lib/tests/test_patch_spec.py
@@ -349,6 +349,7 @@ def test_round_trip_with_patches_on_class(self):
         """Test that extracted patches can be re-applied."""
         original = f"""import unittest
 
+@unittest.skipIf(a == b, "{COMMENT}")
 @unittest.expectedFailure  # {COMMENT}
 class TestFoo(unittest.TestCase):
     ...
@@ -367,6 +368,7 @@ def test_one(self):
 
         # Should have the decorator
         self.assertIn("@unittest.expectedFailure", result)
+        self.assertIn("@unittest.skipIf", result)
         self.assertIn(COMMENT, result)
 
 
