From cb7bee559c5637cc22644883a1040a2ce4d251cc Mon Sep 17 00:00:00 2001 From: Shane Smiskol Date: Sun, 3 Sep 2023 16:25:09 -0700 Subject: [PATCH] use constant --- selfdrive/car/tests/test_models.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/selfdrive/car/tests/test_models.py b/selfdrive/car/tests/test_models.py index e68443566c..462ec2d114 100755 --- a/selfdrive/car/tests/test_models.py +++ b/selfdrive/car/tests/test_models.py @@ -24,6 +24,7 @@ from panda.tests.libpanda import libpanda_py from openpilot.selfdrive.test.helpers import SKIP_ENV_VAR PandaType = log.PandaState.PandaType +SafetyModel = car.CarParams.SafetyModel NUM_JOBS = int(os.environ.get("NUM_JOBS", "1")) JOB_ID = int(os.environ.get("JOB_ID", "0")) @@ -112,11 +113,12 @@ class TestCarModelBase(unittest.TestCase): left_elm = False for msg in lr: if msg.which() == 'pandaStateDEPRECATED': - if msg.pandaStateDEPRECATED.safetyModel.raw != 3: + if msg.pandaStateDEPRECATED.safetyModel != SafetyModel.elm327: left_elm = True + elif msg.which() == 'pandaStates': for ps in msg.pandaStates: - if ps.safetyModel.raw != 3: + if ps.safetyModel != SafetyModel.elm327: left_elm = True break