Files
IQ.Pilot/iqpilot/selfdrive/test/process_replay/test_migration.py
2026-09-03 18:23:24 -05:00

20 lines
710 B
Python

"""
Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos
"""
from iqpilot.cereal import messaging
from iqpilot.selfdrive.test.process_replay.migration import migrate_drivingModelData
def test_driving_model_migration_ignores_incomplete_lane_metadata():
msg = messaging.new_message("modelV2")
msg.modelV2.init("laneLines", 4)
for lane_line in msg.modelV2.laneLines:
lane_line.y = [1.0]
msg.modelV2.laneLineProbs = [0.5]
_, added, _ = migrate_drivingModelData([(0, msg.as_reader())])
assert len(added) == 1
assert added[0].drivingModelData.laneLineMeta.leftProb == 0.0
assert added[0].drivingModelData.laneLineMeta.rightProb == 0.0