mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-03 22:53:44 +08:00
firestarsdog
116 lines
4.6 KiB
Python
116 lines
4.6 KiB
Python
from openpilot.selfdrive.ui.onroad.starpilot.source_bubble_layout import (
|
||
enabled_source_titles,
|
||
fit_source_label,
|
||
source_abbreviated_value_text,
|
||
source_content_metrics,
|
||
source_value_text,
|
||
visible_source_rows,
|
||
)
|
||
|
||
|
||
def test_source_content_metrics_scale_with_visible_row_count():
|
||
assert source_content_metrics(3) == (30, 34, 7)
|
||
assert source_content_metrics(4) == (30, 32, 7)
|
||
assert source_content_metrics(5) == (28, 30, 6)
|
||
|
||
|
||
def test_fit_source_label_preserves_a_safe_value_column_gap():
|
||
def width(text: str) -> int:
|
||
return len(text) * 10
|
||
|
||
assert fit_source_label("Dashboard", "Dash", 80, width) == "Dash"
|
||
assert fit_source_label("Vision", "Vision", 70, width) == "Vision"
|
||
assert fit_source_label("Dashboard", "Dash", 20, width) == "D…"
|
||
|
||
|
||
def test_source_value_text_keeps_missing_values_as_a_dash():
|
||
assert source_value_text(0) == "–"
|
||
assert source_value_text(0.1) == "–"
|
||
assert source_value_text(55) == "55"
|
||
assert source_value_text(float("nan")) == "–"
|
||
assert source_value_text(float("inf")) == "–"
|
||
assert source_abbreviated_value_text(0) == "X"
|
||
assert source_abbreviated_value_text(55) == "55"
|
||
|
||
|
||
def test_enabled_source_titles_follow_priority_and_fallback_settings():
|
||
assert enabled_source_titles(
|
||
"Map Data", "Vision", vision_enabled=True, mapbox_enabled=False,
|
||
) == ("Map Data", "Vision", "Upcoming")
|
||
assert enabled_source_titles(
|
||
"Dashboard", "None", vision_enabled=False, mapbox_enabled=True,
|
||
) == ("Dashboard", "Mapbox")
|
||
assert enabled_source_titles(
|
||
"Highest", "None", vision_enabled=True, mapbox_enabled=True,
|
||
) == ("Dashboard", "Map Data", "Mapbox", "Upcoming")
|
||
assert enabled_source_titles(
|
||
"Dashboard", "Map Data", vision_enabled=False, mapbox_enabled=False,
|
||
dashboard_available=False,
|
||
) == ("Map Data", "Upcoming")
|
||
|
||
|
||
def test_visible_source_rows_honor_active_only_and_source_order():
|
||
source_defs = [
|
||
("Dashboard", "Dash", "dashboard", "Dashboard", "dashboard"),
|
||
("Map Data", "MapD", "map", "Map Data", "map"),
|
||
("Vision", "Vision", "vision", "Vision", "camera"),
|
||
("Mapbox", "MapB", "mapbox", "Mapbox", "map"),
|
||
("Upcoming", "Next", "next", "Next", "next"),
|
||
]
|
||
values = {"dashboard": 45.0, "map": 0.0, "vision": 50.0, "mapbox": 30.0, "next": 20.0}
|
||
|
||
# Map Data has value 0.0, so it is omitted; Dashboard (45.0) is active
|
||
assert visible_source_rows(
|
||
source_defs, values, "Dashboard", ("Dashboard", "Map Data"),
|
||
) == [
|
||
("Dashboard", "dashboard", 45.0, True),
|
||
]
|
||
# When Map Data is the active target but has 0.0 reading, Dashboard is inactive (available standby)
|
||
assert visible_source_rows(
|
||
source_defs, values, "Map Data", ("Dashboard", "Map Data"),
|
||
) == [
|
||
("Dashboard", "dashboard", 45.0, False),
|
||
]
|
||
# Multiple available sources with readings appear in canonical order
|
||
assert visible_source_rows(
|
||
source_defs, values, "Vision", ("Dashboard", "Map Data", "Vision"),
|
||
) == [
|
||
("Dashboard", "dashboard", 45.0, False),
|
||
("Vision", "camera", 50.0, True),
|
||
]
|
||
# When no sources have a valid speed reading (> 0), returns empty list (triggers empty state)
|
||
assert visible_source_rows(
|
||
source_defs, {key: 0.0 for key in values}, "Map Data", ("Map Data",),
|
||
) == []
|
||
|
||
|
||
def test_source_label_color_override_and_engagement_states():
|
||
from openpilot.selfdrive.ui.onroad.starpilot.slc_speed_limit import _source_label_color
|
||
from openpilot.selfdrive.ui.onroad.hud_renderer import COLORS
|
||
from openpilot.selfdrive.ui.ui_state import ui_state, UIStatus
|
||
|
||
# Engaged and not overridden -> Active green
|
||
ui_state.status = UIStatus.ENGAGED
|
||
color = _source_label_color(255, is_overridden=False)
|
||
assert (color.r, color.g, color.b, color.a) == (COLORS.ENGAGED.r, COLORS.ENGAGED.g, COLORS.ENGAGED.b, 255)
|
||
|
||
# Engaged but overridden -> Disengaged/override gray
|
||
color_overridden = _source_label_color(255, is_overridden=True)
|
||
assert (color_overridden.r, color_overridden.g, color_overridden.b, color_overridden.a) == (
|
||
COLORS.DISENGAGED.r, COLORS.DISENGAGED.g, COLORS.DISENGAGED.b, 255
|
||
)
|
||
|
||
# Disengaged -> Disengaged/override gray
|
||
ui_state.status = UIStatus.DISENGAGED
|
||
color_disengaged = _source_label_color(255, is_overridden=False)
|
||
assert (color_disengaged.r, color_disengaged.g, color_disengaged.b, color_disengaged.a) == (
|
||
COLORS.DISENGAGED.r, COLORS.DISENGAGED.g, COLORS.DISENGAGED.b, 255
|
||
)
|
||
|
||
# Override UI status -> Disengaged/override gray
|
||
ui_state.status = UIStatus.OVERRIDE
|
||
color_ui_override = _source_label_color(255, is_overridden=False)
|
||
assert (color_ui_override.r, color_ui_override.g, color_ui_override.b, color_ui_override.a) == (
|
||
COLORS.OVERRIDE.r, COLORS.OVERRIDE.g, COLORS.OVERRIDE.b, 255
|
||
)
|