mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-08 13:33:42 +08:00
firestar5683
64abcb4094
Wire controller callbacks into the running process and add joined planner and mode-action regression coverage.
39 lines
1.5 KiB
Python
39 lines
1.5 KiB
Python
from openpilot.common.test import OpenpilotTestCase
|
|
from openpilot.common.parameterized import parameterized
|
|
|
|
from openpilot.cereal import log
|
|
from opendbc.car.structs import car
|
|
from opendbc.car.car_helpers import interfaces
|
|
from opendbc.car.toyota.values import CAR as TOYOTA
|
|
from opendbc.car.vehicle_model import VehicleModel
|
|
from openpilot.common.realtime import DT_CTRL
|
|
from openpilot.selfdrive.controls.lib.latcontrol_torque import LatControlTorque, LAT_ACCEL_REQUEST_BUFFER_SECONDS
|
|
|
|
def get_controller(car_name):
|
|
CarInterface = interfaces[car_name]
|
|
CP = CarInterface.get_non_essential_params(car_name)
|
|
CI = CarInterface(CP)
|
|
VM = VehicleModel(CP)
|
|
controller = LatControlTorque(CP.as_reader(), CI, DT_CTRL)
|
|
return controller, VM
|
|
|
|
class TestLatControlTorqueBuffer(OpenpilotTestCase):
|
|
|
|
@parameterized.expand([(TOYOTA.TOYOTA_RAV4_TSS2,)])
|
|
def test_request_buffer_consistency(self, car_name):
|
|
buffer_steps = int(LAT_ACCEL_REQUEST_BUFFER_SECONDS / DT_CTRL)
|
|
controller, VM = get_controller(car_name)
|
|
|
|
CS = car.CarState.new_message()
|
|
CS.vEgo = 30
|
|
CS.steeringPressed = False
|
|
params = log.VehicleParameters.new_message()
|
|
|
|
for _ in range(buffer_steps):
|
|
controller.update(True, CS, VM, params, False, 0.001, False, 0.2)
|
|
assert all(val != 0 for val in controller.lat_accel_request_buffer)
|
|
|
|
for _ in range(buffer_steps):
|
|
controller.update(False, CS, VM, params, False, 0.0, False, 0.2)
|
|
assert all(val == 0 for val in controller.lat_accel_request_buffer)
|