mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-08 13:33:42 +08:00
firestarsdog
108 lines
4.8 KiB
Python
108 lines
4.8 KiB
Python
from dataclasses import replace
|
|
from unittest.mock import Mock, patch
|
|
|
|
import pyray as rl
|
|
import pytest
|
|
|
|
from openpilot.starpilot.ui.large_speed_geometry import LARGE_SIGN_SCALE
|
|
from openpilot.starpilot.ui.onroad_state import speed_limit_bounds, slc_controls, compact_sign_obscured_by_actions, SpeedLimitObservation, ObservationKind
|
|
from openpilot.starpilot.ui.presentation import Profile
|
|
from openpilot.starpilot.ui.speed_limit_widget import SpeedLimitWidget
|
|
from openpilot.starpilot.ui.tests.test_unified_speed_native import fonts, state as pending_state
|
|
from openpilot.starpilot.ui.unified_speed_presentation import resolve_unified_speed
|
|
|
|
|
|
def state(**changes):
|
|
current = pending_state(**changes)
|
|
return replace(current, speed_limit=replace(current.speed_limit, pending_speed_limit_mps=None))
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def historical_outline():
|
|
with patch.object(rl, 'draw_rectangle_rounded_lines_ex') as outline:
|
|
yield outline
|
|
|
|
|
|
@pytest.mark.parametrize('profile', list(Profile))
|
|
@pytest.mark.parametrize('offset', [False, True])
|
|
@pytest.mark.parametrize('vienna', [False, True])
|
|
def test_both_sign_styles_keep_their_height_with_or_without_offset(profile, offset, vienna, historical_outline):
|
|
current = state(show_slc_offset=offset)
|
|
current = replace(current, appearance=replace(current.appearance, use_vienna_sign=vienna))
|
|
widget = SpeedLimitWidget(fonts(profile), profile)
|
|
widget.shapes = Mock()
|
|
with patch('openpilot.starpilot.ui.speed_limit_widget.rl.draw_rectangle_rounded') as fill:
|
|
widget.render(current)
|
|
fill.assert_not_called()
|
|
scale = LARGE_SIGN_SCALE if profile == Profile.LARGE else 1
|
|
bounds = speed_limit_bounds(profile, current)
|
|
tall = speed_limit_bounds(profile, replace(current, show_slc_offset=True))
|
|
assert bounds is not None and tall is not None
|
|
assert bounds == tall
|
|
if vienna:
|
|
widget.shapes.outline.assert_not_called()
|
|
widget.shapes.circle.assert_called_once()
|
|
historical_outline.assert_not_called()
|
|
else:
|
|
historical_outline.assert_not_called()
|
|
widget.shapes.outline.assert_called_once()
|
|
assert widget.shapes.outline.call_args.args[:2] == (bounds[0]+12+59*scale, bounds[1]+12+66*scale)
|
|
assert widget.shapes.outline.call_args.kwargs == {}
|
|
assert ('+5' in [call.args[0] for call in widget.fonts.draw.call_args_list]) == offset
|
|
|
|
|
|
@pytest.mark.parametrize('value', (65, 120))
|
|
@pytest.mark.parametrize('metric', (False, True))
|
|
@pytest.mark.parametrize('show', (False, True))
|
|
@pytest.mark.parametrize('offset', (None, 0., .01, -.01, 5., -10.))
|
|
def test_small_sign_keeps_its_footprint_for_every_offset(value, metric, show, offset, historical_outline):
|
|
factor = 3.6 if metric else 2.2369362921
|
|
current = state(metric=metric, show_slc_offset=show)
|
|
current = replace(current, appearance=replace(current.appearance, show_speed_limit_sign=True, use_vienna_sign=False),
|
|
speed_limit=SpeedLimitObservation(ObservationKind.VALID, 'vision', value / factor,
|
|
offset_mps=None if offset is None else offset / factor, action_enabled=False))
|
|
shown = resolve_unified_speed(current)
|
|
widget = SpeedLimitWidget(fonts(Profile.COMPACT), Profile.COMPACT)
|
|
widget.shapes = Mock()
|
|
widget.render(current)
|
|
bounds = speed_limit_bounds(Profile.COMPACT, current)
|
|
assert bounds is not None
|
|
historical_outline.assert_not_called()
|
|
widget.shapes.outline.assert_called_once()
|
|
assert bounds[3] - bounds[1] == 152
|
|
texts = [call.args[0] for call in widget.fonts.draw.call_args_list]
|
|
assert texts == ['VISION', str(value)] + ([shown.offset_text] if shown.offset_text else [])
|
|
|
|
|
|
def test_small_offset_layout_transitions_reuse_the_same_widget(historical_outline):
|
|
widget = SpeedLimitWidget(fonts(Profile.COMPACT), Profile.COMPACT)
|
|
widget.shapes = Mock()
|
|
for enabled in (False, True, False):
|
|
current = state(show_slc_offset=enabled)
|
|
widget.render(current)
|
|
historical_outline.assert_not_called()
|
|
bounds = speed_limit_bounds(Profile.COMPACT, current)
|
|
assert bounds is not None
|
|
assert bounds[3] - bounds[1] == 152
|
|
|
|
|
|
@pytest.mark.parametrize('y', [20, 50, 108])
|
|
def test_attached_compact_chooser_never_hides_its_own_sign(y):
|
|
current = pending_state(slc_prompt_open=True)
|
|
current.customization['layouts']['compact']['speed_limit'].update(x=280 if y == 108 else 330, y=y)
|
|
obscured = compact_sign_obscured_by_actions(current)
|
|
assert obscured is False
|
|
assert (speed_limit_bounds(Profile.COMPACT, current) is None) is obscured
|
|
assert slc_controls(Profile.COMPACT, current)
|
|
|
|
|
|
def test_small_rectangular_no_offset_regression(historical_outline):
|
|
current = state(show_slc_offset=False)
|
|
widget = SpeedLimitWidget(fonts(Profile.COMPACT), Profile.COMPACT)
|
|
widget.shapes = Mock()
|
|
widget.render(current)
|
|
bounds = speed_limit_bounds(Profile.COMPACT, current)
|
|
assert bounds is not None and bounds[3] - bounds[1] == 152
|
|
historical_outline.assert_not_called()
|
|
widget.shapes.outline.assert_called_once()
|