From 4c0e2926e4da2b0c5875e2d8d0dac2c9f9561565 Mon Sep 17 00:00:00 2001 From: Shane Smiskol Date: Fri, 1 Nov 2024 18:30:01 -0500 Subject: [PATCH] fix test_car_model.py segment format (#33915) fix test_car_model.py --- tools/car_porting/test_car_model.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/tools/car_porting/test_car_model.py b/tools/car_porting/test_car_model.py index 6f274aaf69..7ed7ba4f4d 100755 --- a/tools/car_porting/test_car_model.py +++ b/tools/car_porting/test_car_model.py @@ -5,7 +5,7 @@ import unittest # noqa: TID251 from opendbc.car.tests.routes import CarTestRoute from openpilot.selfdrive.car.tests.test_models import TestCarModel -from openpilot.tools.lib.route import SegmentName +from openpilot.tools.lib.route import SegmentRange def create_test_models_suite(routes: list[CarTestRoute], ci=False) -> unittest.TestSuite: @@ -29,10 +29,9 @@ if __name__ == "__main__": parser.print_help() sys.exit() - route_or_segment_name = SegmentName(args.route_or_segment_name.strip(), allow_route_name=True) - segment_num = route_or_segment_name.segment_num if route_or_segment_name.segment_num != -1 else None + sr = SegmentRange(args.route_or_segment_name) - test_route = CarTestRoute(route_or_segment_name.route_name.canonical_name, args.car, segment=segment_num) - test_suite = create_test_models_suite([test_route], ci=args.ci) + test_routes = [CarTestRoute(sr.route_name, args.car, segment=seg_idx) for seg_idx in sr.seg_idxs] + test_suite = create_test_models_suite(test_routes, ci=args.ci) unittest.TextTestRunner().run(test_suite)