def test_validate_non_existing_zone():
zones_at_k_value = {1: ["Zone1", "Zone2"], 3: ["Zone3"]}
zone_map = ZoneMap(zones_at_k_value)
point = TrajectoryPoint(0, 1, 2, 3, "non existing")
point.set_ijk((0, 0, 1))
point.validate_zone(zone_map)
assert not point.valid_zone
def test_validate_zone(k_value, expected_validation):
zones_at_k_value = {1: ["Zone1", "Zone2"], 3: ["Zone3"]}
zone_map = ZoneMap(zones_at_k_value)
point = TrajectoryPoint(0, 1, 2, 3, "Zone1")
point.set_ijk([0, 0, k_value])
assert not point.valid_zone
point.validate_zone(zone_map)
assert point.valid_zone == expected_validation
