Catch warning in test

pydantic · Feb 29, 2024 · 3702dfb · 3702dfb
1 parent 90fb70b
commit 3702dfb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_deprecated_fields.py b/tests/test_deprecated_fields.py
@@ -1,4 +1,5 @@
 import importlib.metadata
+import warnings
 
 import pytest
 from packaging.version import Version
@@ -83,7 +84,10 @@ class Model(BaseModel):
 
         @model_validator(mode='after')
         def validate_x(self) -> Self:
-            self.x = self.x * 2
+            with warnings.catch_warnings():
+                warnings.simplefilter('error', DeprecationWarning)
+                self.x = self.x * 2
+            return self
 
     instance = Model(x=1)