mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-03 06:33:50 +08:00
firestar5683
f5f846f5b6
Co-Authored-By: Prabhaav Pillai <143428353+prabhaavp@users.noreply.github.com>
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import pytest
|
|
|
|
from openpilot.starpilot.system.the_galaxy.the_galaxy import _decode_json_object, _normalize_vasm_config
|
|
|
|
|
|
def test_normalize_vasm_config_accepts_bounded_polygons():
|
|
config = _normalize_vasm_config({
|
|
"width": 1920,
|
|
"height": 1080,
|
|
"poly_left": [[1.2, 2.7], [100, 3], [90, 200]],
|
|
"poly_right": [],
|
|
})
|
|
|
|
assert config == {
|
|
"width": 1920,
|
|
"height": 1080,
|
|
"poly_left": [[1, 3], [100, 3], [90, 200]],
|
|
"poly_right": [],
|
|
}
|
|
|
|
|
|
@pytest.mark.parametrize("config", (
|
|
{},
|
|
{"width": 1920, "height": 1080, "poly_left": [], "poly_right": []},
|
|
{"width": 1920, "height": 1080, "poly_left": [[-1, 1], [2, 2], [3, 3]], "poly_right": []},
|
|
{"width": 1920, "height": 1080, "poly_left": [[1, 1], [2, 2]], "poly_right": []},
|
|
))
|
|
def test_normalize_vasm_config_rejects_unsafe_config(config):
|
|
with pytest.raises(ValueError):
|
|
_normalize_vasm_config(config)
|
|
|
|
|
|
@pytest.mark.parametrize("raw", (
|
|
'{"width":1928,"height":1208,"poly_left":[],"poly_right":[]}',
|
|
b'{"width":1928,"height":1208,"poly_left":[],"poly_right":[]}',
|
|
))
|
|
def test_decode_vasm_config_from_legacy_params_payload(raw):
|
|
assert _decode_json_object(raw) == {
|
|
"width": 1928,
|
|
"height": 1208,
|
|
"poly_left": [],
|
|
"poly_right": [],
|
|
}
|
|
|
|
|
|
@pytest.mark.parametrize("raw", (None, "", "invalid", [], 1))
|
|
def test_decode_vasm_config_rejects_non_objects(raw):
|
|
assert _decode_json_object(raw) == {}
|