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