Files
StarPilot/openpilot/starpilot/longitudinal/tests/extension_helpers.py
T
firestar5683 fc73a3383e Planner Wiring
Wire longitudinal feature selection and inputs into planner hosts after their owners are available.

Co-authored-by: Danny <dannyirwin31@yahoo.com>

Co-authored-by: whoisdomi <dcantu83@gmail.com>
2026-10-14 20:02:39 -05:00

31 lines
1.2 KiB
Python

"""Shared fixtures for the generic longitudinal boundary."""
from openpilot.starpilot.longitudinal.inputs import LongitudinalInputs
def extension_state(controller, name):
return getattr(controller.extension, name, None)
def attach_inputs(controls):
from openpilot.starpilot.vehicle_preferences import VehicleStartupPreferences
controls.vehicle_startup_preferences = VehicleStartupPreferences()
controls.turn_assist_enabled = lambda: False
inputs = LongitudinalInputs.__new__(LongitudinalInputs)
inputs.messages = lambda: controls.sm
inputs.takeoff_preferences = None
inputs.faster_takeoffs = False
for name in ('ev9_long_enabled', 'blended_longitudinal_enabled', 'toyota_sienna_replay', 'toyota_sienna_production', 'toyota_corolla_stop',
'ioniq6_start_enabled', 'gm_start_enabled', 'gm_volt_enabled', 'gm_euv_enabled',
'gm_cc_enabled', 'gm_cc_stop_enabled', 'gm_ascm_enabled', 'gm_suburban_enabled'):
setattr(inputs, name, False)
inputs.gm_cc_evidence = None
controls.longitudinal_inputs = inputs
def empty_params(test):
import tempfile
from openpilot.common.params import Params
directory = tempfile.TemporaryDirectory()
test.addCleanup(directory.cleanup)
return Params(directory.name)