You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.9 KiB
91 lines
2.9 KiB
#!/usr/bin/env python3
|
|
import os
|
|
import unittest
|
|
|
|
import cereal.messaging as messaging
|
|
from cereal.visionipc import VisionIpcClient, VisionStreamType
|
|
from selfdrive.manager.process_config import managed_processes
|
|
|
|
LLK_DECIMATION = 10
|
|
CACHE_PATH = "/data/mbgl-cache-navd.db"
|
|
|
|
def gen_llk():
|
|
msg = messaging.new_message('liveLocationKalman')
|
|
msg.liveLocationKalman.positionGeodetic = {'value': [32.7174, -117.16277, 0], 'std': [0., 0., 0.], 'valid': True}
|
|
msg.liveLocationKalman.calibratedOrientationNED = {'value': [0., 0., 0.], 'std': [0., 0., 0.], 'valid': True}
|
|
msg.liveLocationKalman.status = 'valid'
|
|
return msg
|
|
|
|
|
|
class TestMapRenderer(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
assert "MAPBOX_TOKEN" in os.environ
|
|
|
|
def setUp(self):
|
|
self.sm = messaging.SubMaster(['mapRenderState'])
|
|
self.pm = messaging.PubMaster(['liveLocationKalman'])
|
|
self.vipc = VisionIpcClient("navd", VisionStreamType.VISION_STREAM_MAP, True)
|
|
|
|
if os.path.exists(CACHE_PATH):
|
|
os.remove(CACHE_PATH)
|
|
|
|
def tearDown(self):
|
|
managed_processes['mapsd'].stop()
|
|
|
|
def _run_test(self, valid):
|
|
# start + sync up
|
|
managed_processes['mapsd'].start()
|
|
for _ in range(100):
|
|
self.pm.send("liveLocationKalman", gen_llk())
|
|
self.sm.update(100)
|
|
if self.sm.updated['mapRenderState']:
|
|
break
|
|
assert self.sm.updated['mapRenderState'], "renderer didn't start"
|
|
assert VisionIpcClient.available_streams("navd", False) == {VisionStreamType.VISION_STREAM_MAP, }
|
|
|
|
# connect to vipc
|
|
assert self.vipc.connect(False)
|
|
self.vipc.recv()
|
|
|
|
# run test
|
|
for i in range(20*LLK_DECIMATION):
|
|
frame_expected = (i+1) % LLK_DECIMATION == 0
|
|
prev_frame_id = self.sm['mapRenderState'].frameId
|
|
|
|
llk = gen_llk()
|
|
self.pm.send("liveLocationKalman", llk)
|
|
self.sm.update(200 if frame_expected else 10)
|
|
assert self.sm.updated['mapRenderState'] == frame_expected, "renderer running at wrong frequency"
|
|
|
|
if not frame_expected:
|
|
continue
|
|
|
|
# check output
|
|
assert self.sm.valid['mapRenderState'] == valid
|
|
assert self.sm['mapRenderState'].frameId == (prev_frame_id + 1)
|
|
assert self.sm['mapRenderState'].locationMonoTime == llk.logMonoTime
|
|
if not valid:
|
|
assert self.sm['mapRenderState'].renderTime == 0.
|
|
else:
|
|
assert 0. < self.sm['mapRenderState'].renderTime < 0.1
|
|
|
|
# check vision ipc output
|
|
assert self.vipc.recv() is not None
|
|
assert self.vipc.valid == valid
|
|
assert self.vipc.timestamp_sof == llk.logMonoTime
|
|
assert self.vipc.frame_id == self.sm['mapRenderState'].frameId
|
|
|
|
def test_with_internet(self):
|
|
self._run_test(True)
|
|
|
|
def test_with_no_internet(self):
|
|
token = os.environ['MAPBOX_TOKEN']
|
|
try:
|
|
os.environ['MAPBOX_TOKEN'] = 'invalid_token'
|
|
self._run_test(False)
|
|
finally:
|
|
os.environ['MAPBOX_TOKEN'] = token
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|
|
|