mirror of
https://gitlvb.teallvbs.xyz/IQ.Lvbs/IQ.Pilot.git
synced 2026-07-24 04:52:13 +08:00
IQ.Lvbs CI [bot]
32 lines
1.3 KiB
Python
32 lines
1.3 KiB
Python
import numpy as np
|
|
import pytest
|
|
|
|
from openpilot.selfdrive.controls.lib.longitudinal_mpc_lib.long_mpc import T_IDXS
|
|
from openpilot.selfdrive.controls.lib.longitudinal_planner import get_e2e_accel
|
|
|
|
|
|
def model_velocity(v_ego, v_future):
|
|
return np.interp(T_IDXS, [T_IDXS[0], T_IDXS[-1]], [v_ego, v_future])
|
|
|
|
|
|
class TestE2eCruiseConvergence:
|
|
def test_converges_when_model_wants_to_accelerate(self):
|
|
assert get_e2e_accel(20.0, 30.0, model_velocity(20.0, 25.0), 0.1, False) == pytest.approx(0.5)
|
|
|
|
def test_scales_down_near_cruise_speed(self):
|
|
assert get_e2e_accel(28.5, 30.0, model_velocity(28.5, 30.0), 0.0, False) == pytest.approx(0.05)
|
|
|
|
def test_preserves_active_model_deceleration(self):
|
|
assert get_e2e_accel(20.0, 30.0, model_velocity(20.0, 25.0), -0.05, False) == pytest.approx(-0.05)
|
|
|
|
def test_preserves_future_model_slowdown(self):
|
|
assert get_e2e_accel(20.0, 30.0, model_velocity(20.0, 18.0), 0.1, False) == pytest.approx(0.1)
|
|
|
|
@pytest.mark.parametrize("v_ego, v_cruise, should_stop", [
|
|
(30.0, 30.0, False),
|
|
(31.0, 30.0, False),
|
|
(20.0, 30.0, True),
|
|
])
|
|
def test_never_overrides_cruise_or_stop(self, v_ego, v_cruise, should_stop):
|
|
assert get_e2e_accel(v_ego, v_cruise, model_velocity(v_ego, v_ego + 5.0), -0.2, should_stop) == pytest.approx(-0.2)
|