manual-sync chart-synthesis 2026-07-02T22:45:08Z workspace/tests/test_tangent_chart_synthesis.py

workspace/tests/test_tangent_chart_synthesis.py

@@ -0,0 +1,80 @@
+from __future__ import annotations
+
+from dovla_cil.generation.tangent_chart_synthesis import (
+    ChartSynthesisConfig,
+    _select_chart_proposals,
+)
+from dovla_cil.generation.tangent_memory import TangentExample
+from dovla_cil.generation.tangent_spline_cvae import SplineTangentRow
+
+
+def test_chart_synthesis_config_validates_parameters() -> None:
+    ChartSynthesisConfig(
+        neighbor_pool=8,
+        direct_count=2,
+        barycentric_windows=(2, 4),
+        barycentric_temperature=0.5,
+        utility_weight=0.1,
+        order="interleave",
+    )
+
+    for kwargs, expected in [
+        ({"neighbor_pool": 0}, "neighbor_pool"),
+        ({"direct_count": -1}, "direct_count"),
+        ({"barycentric_windows": (0,)}, "barycentric_windows"),
+        ({"barycentric_temperature": 0.0}, "barycentric_temperature"),
+        ({"utility_weight": -0.1}, "utility_weight"),
+        ({"order": "bad"}, "order"),
+    ]:
+        try:
+            ChartSynthesisConfig(**kwargs)
+        except ValueError as exc:
+            assert expected in str(exc)
+        else:  # pragma: no cover
+            raise AssertionError(f"{expected} should be validated")
+
+
+def test_chart_synthesis_interleaves_direct_and_barycentric_proposals() -> None:
+    query = _row("query", [0.0, 0.0], [0.0, 0.0])
+    train = [
+        _row("a", [0.0, 0.0], [1.0, 0.0]),
+        _row("b", [0.1, 0.0], [3.0, 0.0]),
+        _row("c", [1.0, 0.0], [9.0, 0.0]),
+    ]
+    config = ChartSynthesisConfig(
+        neighbor_pool=3,
+        direct_count=1,
+        barycentric_windows=(2,),
+        barycentric_temperature=1.0,
+        order="interleave",
+    )
+
+    proposals = _select_chart_proposals(
+        query,
+        train_positive=train,
+        train_positive_by_task={"task": train},
+        config=config,
+        max_k=3,
+    )
+
+    assert proposals[0] == [1.0, 0.0]
+    assert 1.0 < proposals[1][0] < 3.0
+    assert proposals[2] == [3.0, 0.0]
+
+
+def _row(group_id: str, condition: list[float], delta: list[float]) -> SplineTangentRow:
+    return SplineTangentRow(
+        example=TangentExample(
+            group_id=group_id,
+            task_id="task",
+            candidate_type="candidate",
+            tangent_label="positive",
+            delta_utility=1.0,
+            delta_action=[delta],
+        ),
+        condition=condition,
+        code=delta,
+        full_delta=delta,
+        horizon=1,
+        action_dim=len(delta),
+    )