Parameterized & Data-Driven Tests

"""Parametrized AEB tests: same test logic, different signal values."""
import pytest

AEB_TEST_CASES = [
    # (speed_kmh, radar_dist_m, expected_aeb_state, expected_brake_min)
    (10,  40,  "INACTIVE",  0),    # too slow for AEB
    (50,  40,  "ACTIVE",   30),    # normal AEB activation
    (80,  35,  "ACTIVE",   60),    # higher speed = more braking
    (120, 30,  "ACTIVE",   90),    # highway AEB
    (200, 40,  "ACTIVE",   30),    # above ODD - AEB may not activate
    (50,  100, "INACTIVE",  0),    # target too far
    (50,   5,  "ACTIVE",  100),    # imminent collision
]

@pytest.mark.parametrize(
    "speed, radar_dist, expected_state, expected_brake_min",
    AEB_TEST_CASES,
    ids=[f"speed{t[0]}_dist{t[1]}" for t in AEB_TEST_CASES]
)
def test_aeb_response(ecu, speed, radar_dist,
                       expected_state, expected_brake_min):
    ecu.send_signal("VehicleSpeed", speed)
    ecu.send_signal("RadarTarget_Distance", radar_dist)
    import time; time.sleep(0.3)
    state = ecu.read_signal("AEB_State", timeout_s=0.5)
    assert state == expected_state, f"AEB state {state} != {expected_state}"
    if expected_brake_min > 0:
        brake = ecu.read_signal("AEB_BrakeRequest_pct", timeout_s=0.5)
        assert brake >= expected_brake_min, f"Brake {brake} < {expected_brake_min}"

"""Load test parameters from CSV for data-driven testing."""
import pytest, csv
from pathlib import Path

def load_test_vectors(csv_path: str):
    """Load test vectors from CSV; first row is headers."""
    rows = []
    with open(csv_path) as f:
        reader = csv.DictReader(f)
        for row in reader:
            if not row.get("skip", "").lower() == "true":
                rows.append({
                    "speed": float(row["speed_kmh"]),
                    "dist":  float(row["radar_dist_m"]),
                    "exp_state": row["expected_aeb_state"],
                    "exp_brake":  float(row["expected_brake_min_pct"]),
                    "req_id":     row["req_id"],
                    "tc_id":      row["tc_id"],
                })
    return rows

TEST_VECTORS = load_test_vectors("test_data/aeb_vectors.csv")

@pytest.mark.parametrize("vec", TEST_VECTORS,
                          ids=[v["tc_id"] for v in TEST_VECTORS])
def test_aeb_csv_driven(ecu, vec):
    ecu.send_signal("VehicleSpeed", vec["speed"])
    ecu.send_signal("RadarTarget_Distance", vec["dist"])
    import time; time.sleep(0.3)
    state = ecu.read_signal("AEB_State", timeout_s=0.5)
    assert state == vec["exp_state"]