From 824c7abacfbb6c3f36fbcfaad2a07b5fa08fb56f Mon Sep 17 00:00:00 2001 From: Shane Smiskol Date: Thu, 14 Dec 2023 21:42:35 -0800 Subject: [PATCH] get size from dict --- selfdrive/car/tests/test_models.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/selfdrive/car/tests/test_models.py b/selfdrive/car/tests/test_models.py index 7ac49c1c24..15460de4a6 100755 --- a/selfdrive/car/tests/test_models.py +++ b/selfdrive/car/tests/test_models.py @@ -328,12 +328,12 @@ class TestCarModelBase(unittest.TestCase): if self.CP.dashcamOnly: self.skipTest("no need to check panda safety for dashcamOnly") - valid_addrs = [(addr, bus) for bus, addrs in self.fingerprint.items() for addr in addrs] - address, bus = data.draw(st.sampled_from(valid_addrs)) + valid_addrs = [(addr, bus, size) for bus, addrs in self.fingerprint.items() for addr, size in addrs.items()] + address, bus, size = data.draw(st.sampled_from(valid_addrs)) # address = 0x201 # bus = 0 # print('addr, bus:', address, bus) - size = self.fingerprint[bus][address] + # size = self.fingerprint[bus][address] msg_strategy = st.binary(min_size=size, max_size=size) msgs = data.draw(st.lists(msg_strategy, min_size=20))