mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-01 05:33:49 +08:00
firestar5683
55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
import pytest
|
|
|
|
from openpilot.selfdrive.modeld import modeld
|
|
|
|
|
|
class FakeParams:
|
|
def __init__(self):
|
|
self.values = {}
|
|
|
|
def put(self, key, value):
|
|
self.values[key] = value
|
|
|
|
|
|
@pytest.mark.parametrize(("model_output", "dropped_frames", "external_gpu_active", "expected"), [
|
|
(object(), 0, False, True),
|
|
(object(), 1, False, False),
|
|
(object(), 2, False, False),
|
|
(object(), 1, True, True),
|
|
(object(), 2, True, True),
|
|
(None, 0, False, False),
|
|
(None, 1, True, False),
|
|
])
|
|
def test_model_output_is_suppressed_after_vipc_drop(model_output, dropped_frames, external_gpu_active, expected):
|
|
assert modeld._should_publish_model_output(model_output, dropped_frames, external_gpu_active) is expected
|
|
|
|
|
|
def test_incompatible_downloaded_model_falls_back_to_builtin(monkeypatch):
|
|
calls = []
|
|
builtin_model = object()
|
|
|
|
def load_model(cam_w, cam_h, external_gpu_active):
|
|
calls.append((cam_w, cam_h, external_gpu_active))
|
|
if len(calls) == 1:
|
|
raise TypeError("incompatible artifact")
|
|
return builtin_model
|
|
|
|
params = FakeParams()
|
|
monkeypatch.setattr(modeld, "ModelState", load_model)
|
|
monkeypatch.setattr(modeld.cloudlog, "exception", lambda *_args, **_kwargs: None)
|
|
|
|
assert modeld._load_model_state(1928, 1208, "custom-model", False, params) is builtin_model
|
|
assert calls == [(1928, 1208, False), (1928, 1208, False)]
|
|
assert params.values == {
|
|
"Model": modeld.BUILTIN_MODEL_KEY,
|
|
"DrivingModel": modeld.BUILTIN_MODEL_KEY,
|
|
"DrivingModelName": "Regret Driven Framework V4",
|
|
}
|
|
|
|
|
|
def test_builtin_model_load_failure_is_not_hidden(monkeypatch):
|
|
monkeypatch.setattr(modeld, "ModelState", lambda *_args: (_ for _ in ()).throw(TypeError("bad builtin")))
|
|
|
|
with pytest.raises(TypeError, match="bad builtin"):
|
|
modeld._load_model_state(1928, 1208, modeld.BUILTIN_MODEL_KEY, False, FakeParams())
|