144 lines
3.6 KiB
Python
Executable File
144 lines
3.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import os
|
|
import unittest
|
|
|
|
from common.params import Params
|
|
from selfdrive.test.longitudinal_maneuvers.maneuver import Maneuver
|
|
|
|
|
|
# TODO: make new FCW tests
|
|
maneuvers = [
|
|
Maneuver(
|
|
'approach stopped car at 20m/s, initial distance: 120m',
|
|
duration=20.,
|
|
initial_speed=25.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=120.,
|
|
speed_lead_values=[30., 0.],
|
|
breakpoints=[0., 1.],
|
|
),
|
|
Maneuver(
|
|
'approach stopped car at 20m/s, initial distance 90m',
|
|
duration=20.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=90.,
|
|
speed_lead_values=[20., 0.],
|
|
breakpoints=[0., 1.],
|
|
),
|
|
Maneuver(
|
|
'steady state following a car at 20m/s, then lead decel to 0mph at 1m/s^2',
|
|
duration=50.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=35.,
|
|
speed_lead_values=[20., 20., 0.],
|
|
breakpoints=[0., 15., 35.0],
|
|
),
|
|
Maneuver(
|
|
'steady state following a car at 20m/s, then lead decel to 0mph at 2m/s^2',
|
|
duration=50.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=35.,
|
|
speed_lead_values=[20., 20., 0.],
|
|
breakpoints=[0., 15., 25.0],
|
|
),
|
|
Maneuver(
|
|
'steady state following a car at 20m/s, then lead decel to 0mph at 3m/s^2',
|
|
duration=50.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=35.,
|
|
speed_lead_values=[20., 20., 0.],
|
|
breakpoints=[0., 15., 21.66],
|
|
),
|
|
Maneuver(
|
|
'steady state following a car at 20m/s, then lead decel to 0mph at 3+m/s^2',
|
|
duration=40.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=35.,
|
|
speed_lead_values=[20., 20., 0.],
|
|
prob_lead_values=[0., 1., 1.],
|
|
cruise_values=[20., 20., 20.],
|
|
breakpoints=[2., 2.01, 8.8],
|
|
),
|
|
Maneuver(
|
|
"approach stopped car at 20m/s, with prob_lead_values",
|
|
duration=30.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=120.,
|
|
speed_lead_values=[0.0, 0., 0.],
|
|
prob_lead_values=[0.0, 0., 1.],
|
|
cruise_values=[20., 20., 20.],
|
|
breakpoints=[0.0, 2., 2.01],
|
|
),
|
|
Maneuver(
|
|
"approach slower cut-in car at 20m/s",
|
|
duration=20.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=50.,
|
|
speed_lead_values=[15., 15.],
|
|
breakpoints=[1., 11.],
|
|
only_lead2=True,
|
|
),
|
|
Maneuver(
|
|
"stay stopped behind radar override lead",
|
|
duration=20.,
|
|
initial_speed=0.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=10.,
|
|
speed_lead_values=[0., 0.],
|
|
prob_lead_values=[0., 0.],
|
|
breakpoints=[1., 11.],
|
|
only_radar=True,
|
|
),
|
|
Maneuver(
|
|
"NaN recovery",
|
|
duration=30.,
|
|
initial_speed=15.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=60.,
|
|
speed_lead_values=[0., 0., 0.0],
|
|
breakpoints=[1., 1.01, 11.],
|
|
cruise_values=[float("nan"), 15., 15.],
|
|
),
|
|
]
|
|
|
|
|
|
class LongitudinalControl(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
os.environ['SIMULATION'] = "1"
|
|
os.environ['SKIP_FW_QUERY'] = "1"
|
|
os.environ['NO_CAN_TIMEOUT'] = "1"
|
|
|
|
params = Params()
|
|
params.clear_all()
|
|
params.put_bool("Passive", bool(os.getenv("PASSIVE")))
|
|
params.put_bool("OpenpilotEnabledToggle", True)
|
|
|
|
# hack
|
|
def test_longitudinal_setup(self):
|
|
pass
|
|
|
|
|
|
def run_maneuver_worker(k):
|
|
def run(self):
|
|
man = maneuvers[k]
|
|
print(man.title)
|
|
valid, _ = man.evaluate()
|
|
self.assertTrue(valid, msg=man.title)
|
|
return run
|
|
|
|
|
|
for k in range(len(maneuvers)):
|
|
setattr(LongitudinalControl, f"test_longitudinal_maneuvers_{k+1}",
|
|
run_maneuver_worker(k))
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(failfast=True)
|