fix

Author: Miskler

src/openinflation_dataclass/card.py

@@ -88,7 +88,7 @@ class Card(NetworkModel):
 
     @model_validator(mode="after")
     def validate_business_rules(self) -> Card:
-        is_piece_unit = self.unit == "PCE"
+        is_piece_unit = self.unit_net == "PCE"
         has_count = self.available_count is not None
         count_is_int = type(self.available_count) is int
 
@@ -98,8 +98,8 @@ def validate_business_rules(self) -> Card:
             raise ValueError(
                 "package_quantity_net and package_unit must be set together or both set to None."
             )
-        if (self.package_quantity_gross is None) != (self.package_unit is None):
+        if (self.package_weight_gross is None) != (self.package_unit is None):
             raise ValueError(
-                "package_quantity_gross and package_unit must be set together or both set to None."
+                "package_weight_gross and package_unit must be set together or both set to None."
             )
         return self

tests/test_models.py

@@ -40,10 +40,10 @@ def _build_card() -> Card:
         loyal_price=75.9,
         wholesale_price=[],
         price_unit="RUB",
-        unit="PCE",
+        unit_net="PCE",
         available_count=15,
         package_quantity_net=1.0,
-        package_quantity_gross=1.0,
+        package_weight_gross=1.0,
         package_unit="LTR",
         categories_uid=["milk"],
         main_image="main-image.png",
@@ -132,9 +132,10 @@ def test_card_piece_validation_rejects_float_count() -> None:
             loyal_price=75.9,
             wholesale_price=[],
             price_unit="RUB",
-            unit="PCE",
+            unit_net="PCE",
             available_count=15.5,
-            package_quantity=1.0,
+            package_quantity_net=1.0,
+            package_weight_gross=1.0,
             package_unit="LTR",
             categories_uid=["milk"],
             main_image="main-image.png",