vee/Cognitive_Agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

backend

Browse Source
This commit is contained in:
orin
2026-07-30 14:09:04 +08:00
parent 25b24ac4cd
commit 49f5c49c53
35 changed files with 4078 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit/test_mode_profile.py
+265
View File
@@ -0,0 +1,265 @@
from __future__ import annotations
import tempfile
import unittest
from pathlib import Path
import yaml
from cognitive_agent.application.mode_profile import ModeProfileRegistry
from cognitive_agent.application.seat_service import REPOSITORY_ROOT
class ModeProfileRegistryTests(unittest.TestCase):
def test_zero_gravity_profile_matches_alias_and_builds_requests(self) -> None:
registry = ModeProfileRegistry(REPOSITORY_ROOT / "config" / "modes")
matched = registry.match_text("副驾驶进入零重力模式", vehicle_id="denza_n7")
self.assertEqual([profile.id for profile in matched], ["zero_gravity"])
self.assertEqual(matched[0].vehicle_ids, ("denza_n7",))
requests = registry.requests_for_profiles(matched)
self.assertGreaterEqual(len(requests), 4)
self.assertEqual(requests[0]["domain"], "seat")
self.assertEqual(requests[0]["capability"], "seat.set_longitudinal_position")
self.assertEqual(requests[0]["target_scope"], "passenger")
self.assertIn(
{"name": "position_pct", "value": 0},
requests[0]["parameters"],
)
by_capability = {request["capability"]: request for request in requests}
self.assertEqual(
by_capability["seat.set_cushion_tilt"]["parameters"],
[{"name": "position_pct", "value": 100}],
)
self.assertEqual(
by_capability["seat.set_backrest_recline"]["parameters"],
[
{"name": "position_pct", "value": 81},
{"name": "angle_deg", "value": 150},
],
)
self.assertEqual(
by_capability["seat.set_legrest_height"]["parameters"],
[
{"name": "position_pct", "value": 87},
{"name": "angle_deg", "value": 163},
],
)
public = registry.public_modes()
self.assertIn("零重力模式", {item["name"] for item in public})
self.assertNotIn(
"zero_gravity",
{profile.id for profile in registry.match_text("副驾驶进入零重力模式", vehicle_id="other_vehicle")},
)
def test_reference_modes_are_denza_specific(self) -> None:
registry = ModeProfileRegistry(REPOSITORY_ROOT / "config" / "modes")
expected_ids = {
"zero_gravity",
"cruise_sofa",
"leisure_lounger",
"nap",
"easy_entry_exit",
"baby_nap",
"office",
"meditation_space",
"electric_beach_chair",
"queen_passenger_driving",
"calf_decompression_float",
}
denza_modes = registry.public_modes(vehicle_id="denza_n7")
denza_ids = {mode["id"] for mode in denza_modes}
self.assertTrue(expected_ids.issubset(denza_ids))
self.assertTrue(
expected_ids.isdisjoint(
{mode["id"] for mode in registry.public_modes(vehicle_id="other_vehicle")}
)
)
self.assertEqual(
registry.match_text("进入办公模式", vehicle_id="denza_n7")[0].id,
"office",
)
self.assertEqual(
registry.match_text("进入小腿解压悬浮模式", vehicle_id="denza_n7")[0].id,
"calf_decompression_float",
)
def test_common_modes_are_visible_with_vehicle_specific_overrides(self) -> None:
with tempfile.TemporaryDirectory() as directory:
root = Path(directory)
common = root / "common"
denza = root / "vehicles" / "denza_n7"
other = root / "vehicles" / "other_vehicle"
common.mkdir(parents=True)
denza.mkdir(parents=True)
other.mkdir(parents=True)
(common / "relax.yaml").write_text(
"""
id: relax
name: 放松模式
aliases:
- relax
description: 公共放松模式。
operations:
- domain: roof
capability: roof.set_transmittance
target_scope: vehicle
operation: set
parameters:
transmittance_pct: 45
""".strip()
+ "\n",
encoding="utf-8",
)
(denza / "relax.yaml").write_text(
"""
id: relax
name: N7 放松模式
aliases:
- relax
description: N7 专属放松模式。
scope:
vehicle_ids:
- denza_n7
operations:
- domain: roof
capability: roof.set_transmittance
target_scope: vehicle
operation: set
parameters:
transmittance_pct: 20
""".strip()
+ "\n",
encoding="utf-8",
)
(other / "nap.yaml").write_text(
"""
id: nap
name: 午休模式
description: 其他车型专属午休模式。
operations:
- domain: roof
capability: roof.set_transmittance
target_scope: vehicle
operation: set
parameters:
transmittance_pct: 10
""".strip()
+ "\n",
encoding="utf-8",
)
registry = ModeProfileRegistry(root)
denza_matches = registry.match_text("进入放松模式", vehicle_id="denza_n7")
other_matches = registry.match_text("进入放松模式", vehicle_id="other_vehicle")
self.assertEqual([profile.name for profile in denza_matches], ["N7 放松模式"])
self.assertEqual([profile.name for profile in other_matches], ["放松模式"])
self.assertEqual(
{profile.id for profile in registry.match_text("进入午休模式", vehicle_id="denza_n7")},
set(),
)
self.assertEqual(
{mode["id"] for mode in registry.public_modes(vehicle_id="denza_n7")},
{"relax"},
)
def test_save_mode_profile_persists_yaml_and_reloads(self) -> None:
with tempfile.TemporaryDirectory() as directory:
root = Path(directory)
registry = ModeProfileRegistry(root)
profile = registry.save_mode(
"露营模式",
[
{
"domain": "roof",
"capability": "roof.set_transmittance",
"target_scope": "vehicle",
"operation": "set",
"parameters": {"transmittance_pct": 20},
"reason": "保存最近一次天幕目标。",
}
],
vehicle_id="denza_n7",
)
saved_path = Path(profile.source)
self.assertTrue(saved_path.is_file())
self.assertEqual(
saved_path.relative_to(root).parts[:3],
("vehicles", "denza_n7", "user"),
)
document = yaml.safe_load(saved_path.read_text(encoding="utf-8"))
self.assertEqual(document["name"], "露营模式")
self.assertEqual(document["scope"]["vehicle_ids"], ["denza_n7"])
self.assertEqual(document["operations"][0]["domain"], "roof")
reloaded = ModeProfileRegistry(root)
matched = reloaded.match_text("进入露营模式", vehicle_id="denza_n7")
self.assertEqual([item.name for item in matched], ["露营模式"])
self.assertEqual(
reloaded.match_text("进入露营模式", vehicle_id="other_vehicle"),
[],
)
requests = reloaded.requests_for_profiles(matched)
self.assertEqual(requests[0]["capability"], "roof.set_transmittance")
self.assertEqual(
requests[0]["parameters"],
[{"name": "transmittance_pct", "value": 20}],
)
def test_update_mode_profile_backs_up_initial_yaml_and_restores(self) -> None:
with tempfile.TemporaryDirectory() as directory:
root = Path(directory)
registry = ModeProfileRegistry(root)
profile = registry.save_mode(
"露营模式",
[
{
"domain": "roof",
"capability": "roof.set_transmittance",
"target_scope": "vehicle",
"operation": "set",
"parameters": {"transmittance_pct": 20},
"reason": "原始天幕目标。",
}
],
vehicle_id="denza_n7",
)
updated = registry.update_mode(
profile,
[
{
"domain": "roof",
"capability": "roof.set_transmittance",
"target_scope": "vehicle",
"operation": "set",
"parameters": {"transmittance_pct": 50},
"reason": "用户修改天幕目标。",
}
],
)
self.assertTrue(Path(profile.source).with_suffix(".yaml.initial.bak").is_file())
requests = registry.requests_for_profiles([updated])
self.assertEqual(
requests[0]["parameters"],
[{"name": "transmittance_pct", "value": 50}],
)
restored = registry.restore_initial_mode(updated)
restored_requests = registry.requests_for_profiles([restored])
self.assertEqual(
restored_requests[0]["parameters"],
[{"name": "transmittance_pct", "value": 20}],
)
if __name__ == "__main__":
unittest.main()
tests/unit/test_vehicle_profile.py
+72
View File
@@ -0,0 +1,72 @@
from __future__ import annotations
import unittest
from pathlib import Path
from cognitive_agent.application.vehicle_profile import VehicleProfileRegistry
ROOT = Path(__file__).resolve().parents[2]
class VehicleProfileTests(unittest.TestCase):
def setUp(self) -> None:
self.registry = VehicleProfileRegistry(ROOT / "config" / "vehicles")
def test_loads_denza_n7_seat_motor_calibration(self) -> None:
profile = self.registry.get("denza_n7")
self.assertEqual(profile.id, "denza_n7")
self.assertEqual(profile.name, "腾势 N7")
self.assertEqual(set(profile.seat_motors), {
"longitudinal",
"cushion_tilt",
"backrest_recline",
"legrest_height",
})
self.assertEqual(profile.seat_motors["longitudinal"].range, 215)
self.assertEqual(profile.seat_motors["longitudinal"].hall_total_pulses, 480)
self.assertEqual(profile.seat_motors["longitudinal"].reference_zero, "rearmost")
self.assertEqual(profile.seat_motors["backrest_recline"].range, 68)
self.assertEqual(profile.seat_motors["backrest_recline"].hall_total_pulses, 917)
self.assertEqual(profile.seat_motors["legrest_height"].range, 78)
self.assertEqual(profile.seat_motors["legrest_height"].hall_total_pulses, 647)
def test_converts_relative_physical_delta_to_absolute_percent(self) -> None:
profile = self.registry.get("denza_n7")
self.assertEqual(
profile.relative_target_pct("longitudinal", current_pct=50, delta=50, direction="increase"),
73,
)
self.assertEqual(
profile.relative_target_pct("backrest_recline", current_pct=20, delta=10, direction="increase"),
35,
)
self.assertEqual(
profile.relative_target_pct("legrest_height", current_pct=10, delta=10, direction="decrease"),
0,
)
def test_exposes_display_angle_ranges_for_frontend(self) -> None:
profile = self.registry.get("denza_n7")
public = profile.public_dict()
backrest = public["seat"]["motors"]["backrest_recline"]
legrest = public["seat"]["motors"]["legrest_height"]
self.assertEqual(backrest["display_angle"], {
"kind": "included_angle",
"zero_deg": 95,
"max_deg": 163,
"provisional": True,
})
self.assertEqual(legrest["display_angle"], {
"kind": "included_angle",
"zero_deg": 95,
"max_deg": 173,
"provisional": True,
})
if __name__ == "__main__":
unittest.main()