mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-07 13:03:42 +08:00
Dom
63 lines
3.0 KiB
Python
63 lines
3.0 KiB
Python
from contextlib import ExitStack
|
|
from types import SimpleNamespace as NS
|
|
from unittest.mock import Mock, patch
|
|
|
|
import pyray as rl
|
|
|
|
from openpilot.starpilot.conditional_mode.policy import ModeChoice
|
|
from openpilot.starpilot.ui.conditional_status import ConditionalDisplay
|
|
from openpilot.starpilot.ui.onroad import OnroadView, axis_status_color
|
|
from openpilot.starpilot.ui.onroad_cem_gauge import AetherGauge, IndicatorType
|
|
from openpilot.starpilot.ui.onroad_compact_widgets import MiciSidebarWidgets
|
|
from openpilot.starpilot.ui.onroad_conditional import stop_active
|
|
from openpilot.starpilot.ui.onroad_state import OnroadState, SpeedLimitObservation
|
|
from openpilot.starpilot.ui import onroad
|
|
from openpilot.starpilot.ui.presentation import Profile
|
|
|
|
|
|
def aol_light():
|
|
return OnroadState(True, False, 10., 50., SpeedLimitObservation(), lateral_active=True,
|
|
conditional_configured=ModeChoice.CEM,
|
|
conditional_perception=ConditionalDisplay(ModeChoice.CEM, False, 'cem_stop', 8, 'planner', 1))
|
|
|
|
|
|
def test_compact_aol_light_is_informational_without_orange_border():
|
|
state = aol_light()
|
|
widget = MiciSidebarWidgets(Mock())
|
|
with patch.object(widget, '_stop_icon') as light, patch.object(widget, '_chill_icon') as moon:
|
|
widget._conditional(rl.Rectangle(476, 80, 60, 80), state)
|
|
light.assert_called_once()
|
|
moon.assert_not_called()
|
|
assert not stop_active(state)
|
|
assert (axis_status_color(state).r, axis_status_color(state).g, axis_status_color(state).b) == (10, 186, 181)
|
|
|
|
|
|
def test_large_aol_light_uses_same_perception_without_control_authority():
|
|
view = OnroadView.__new__(OnroadView)
|
|
view.fonts = Mock(profile=Profile.LARGE)
|
|
view.fonts.measure.return_value = NS(width=100, height=30)
|
|
view.cem_gauge = AetherGauge(view.fonts)
|
|
view.navigation_favorites = onroad.NavigationFavorites(view.fonts)
|
|
view.camera_layer = view.background_layer = view.extra_overlays = None
|
|
view.projection_viewport = None
|
|
view._corner_cache = onroad.CornerHintCache()
|
|
view.unified_speed = Mock()
|
|
view.unified_speed.bounds.return_value = rl.Rectangle(88, 75, 176, 196)
|
|
view.current_speed = view.steering_wheel = view.torque_bar = Mock()
|
|
view.navigation = view.alert = Mock()
|
|
with ExitStack() as stack:
|
|
for name in dir(rl):
|
|
if name.startswith('draw_'):
|
|
stack.enter_context(patch.object(rl, name))
|
|
stack.enter_context(patch.object(onroad.clip, 'begin_scissor_mode'))
|
|
stack.enter_context(patch.object(onroad.clip, 'end_scissor_mode'))
|
|
stack.enter_context(patch.object(onroad, 'render_corner_hint'))
|
|
stack.enter_context(patch.object(onroad, 'render_glow'))
|
|
light = stack.enter_context(patch.object(view.cem_gauge, '_render_unified_road'))
|
|
state = aol_light()
|
|
view._large(state)
|
|
light.assert_called_once()
|
|
assert light.call_args.args[3].indicator_type is IndicatorType.STOP_LIGHT
|
|
assert not state.longitudinal_active and not stop_active(state)
|
|
assert (axis_status_color(state).r, axis_status_color(state).g, axis_status_color(state).b) == (10, 186, 181)
|