squash: extend test

tests/unit/test_hardware.py

@@ -67,6 +67,26 @@ def test_constraint_name_pattern(value: str, expected: tuple[Any, Any]) -> None:
     assert match.groups() == expected
 
 
+_size_constraint_pattern_input = [
+    ({'name': 'num_with_default', 'raw_value': '10', 'default_unit': 'GiB'},
+     'num_with_default: == 10 gibibyte'),
+    ({'name': 'num_without_default', 'raw_value': '1024'}, 'num_without_default: == 1024 byte'),
+    ({'name': 'num_with_unit', 'raw_value': '10 GiB', 'default_unit': 'GiB'},
+     'num_with_unit: == 10 GiB'),
+    ]
+
+
+@pytest.mark.parametrize(
+    ('value', 'expected'),
+    _size_constraint_pattern_input,
+    )
+def test_constraint_default_unit(value: dict, expected: tuple[Any, Any]) -> None:
+    constraint_out = tmt.hardware.SizeConstraint.from_specification(**value)
+
+    assert constraint_out is not None
+    assert str(constraint_out) == expected
+
+
 _constraint_components_pattern_input = [
     ('memory 10 GiB', ('memory', None, None, None, '10 GiB')),
     ('cpu.processors != 4 ', ('cpu', None, 'processors', '!=', '4')),

tmt/hardware.py

@@ -493,7 +493,7 @@ class Constraint(BaseConstraint, Generic[ConstraintValueT]):
     raw_value: str
 
     # If set, it is a raw unit specified by the constraint.
-    unit: Optional[str] = None
+    default_unit: Optional[str] = None
 
     # If set, it is a "bigger" constraint, to which this constraint logically
     # belongs as one of its aspects.
@@ -508,7 +508,7 @@ def _from_specification(
             as_cast: Optional[Callable[[str], ConstraintValueT]] = None,
             original_constraint: Optional['Constraint[Any]'] = None,
             allowed_operators: Optional[list[Operator]] = None,
-            unit: Optional[Any] = "bytes"
+            default_unit: Optional[Any] = "bytes"
             ) -> T:
         """
         Parse raw constraint specification into our internal representation.
@@ -561,7 +561,7 @@ def _from_specification(
             if not isinstance(
                     value,
                     pint.Quantity):  # type: ignore[reportUnnecessaryIsInstance,unused-ignore]
-                value = pint.Quantity(value, unit)
+                value = pint.Quantity(value, default_unit)
 
         elif as_cast is not None:
             value = as_cast(raw_value)
@@ -674,15 +674,15 @@ def from_specification(
             raw_value: str,
             original_constraint: Optional['Constraint[Any]'] = None,
             allowed_operators: Optional[list[Operator]] = None,
-            unit: Optional[Any] = 'bytes'
+            default_unit: Optional[Any] = 'bytes'
             ) -> T:
         return cls._from_specification(
             name,
             raw_value,
             as_quantity=True,
             original_constraint=original_constraint,
             allowed_operators=allowed_operators,
-            unit=unit
+            default_unit=default_unit
             )