cleanup json_schema_extras

JakobGM · Oct 27, 2023 · 22e0fef · 22e0fef
1 parent e0bf587
commit 22e0fef
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_validators.py b/tests/test_validators.py
@@ -235,7 +235,7 @@ def test_uniqueness_validation():
     """It should be able to validate uniqueness."""
 
     class MyModel(pt.Model):
-        column: int = pt.Field(json_schema_extra={"unique": True})
+        column: int = pt.Field(unique=True)
 
     non_duplicated_df = pt.DataFrame({"column": [1, 2, 3]})
     MyModel.validate(non_duplicated_df)
@@ -346,7 +346,7 @@ def test_uniqueness_constraint_validation():
     """Uniqueness constraints should be validated."""
 
     class UniqueModel(pt.Model):
-        product_id: int = pt.Field(json_schema_extra={"unique": True})
+        product_id: int = pt.Field(unique=True)
 
     validate(dataframe=pl.DataFrame({"product_id": [1, 2]}), schema=UniqueModel)