diff --git a/tests/unit/test_protocol_validator.py b/tests/unit/test_protocol_validator.py index d402433..cd5d67e 100644 --- a/tests/unit/test_protocol_validator.py +++ b/tests/unit/test_protocol_validator.py @@ -153,19 +153,19 @@ def test_feature_unknown_column_fails() -> None: validate_strategy(ast) -def test_validator_accepts_temporal_features() -> None: - for name in ("hour", "dow", "is_weekend", "minute_of_hour"): - src = _wrap( - { - "op": "gt", - "args": [ - {"kind": "feature", "name": name}, - {"kind": "literal", "value": 0}, - ], - } - ) - ast = parse_strategy(src) - validate_strategy(ast) # no exception +@pytest.mark.parametrize("name", ["hour", "dow", "is_weekend", "minute_of_hour"]) +def test_validator_accepts_temporal_feature(name: str) -> None: + src = _wrap( + { + "op": "gt", + "args": [ + {"kind": "feature", "name": name}, + {"kind": "literal", "value": 0.0}, + ], + } + ) + ast = parse_strategy(src) + validate_strategy(ast) # no exception def test_validator_rejects_temporal_typo() -> None: @@ -174,7 +174,7 @@ def test_validator_rejects_temporal_typo() -> None: "op": "gt", "args": [ {"kind": "feature", "name": "weekday"}, - {"kind": "literal", "value": 0}, + {"kind": "literal", "value": 0.0}, ], } )