tenant.openagents/omega

1from __future__ import annotations
2
3import unittest
4
5from zed_eval import baseline
6
7
8class BaselineRecordTests(unittest.TestCase):
9    def test_record_uses_base_ref_from_source_info(self) -> None:
10        record, has_build_info = baseline._record_from_provenance(
11            namespace="alice",
12            fallback_experiment="swe-atlas-rf",
13            run_id="run-1",
14            provenance={
15                "request": {
16                    "experiment_name": "swe-atlas-rf",
17                    "agent_model": "anthropic/claude-sonnet-4-6",
18                    "judge_preset": "kimi-k2.7-code",
19                    "build_id": "bld-test",
20                },
21                "summary": {"status": "completed", "trial_count": 10},
22                "build_info": {
23                    "build_id": "bld-test",
24                    "base_sha": "a" * 40,
25                    "patch_sha256": None,
26                    "source": {"base_ref": "v0.210.0"},
27                },
28            },
29        )
30
31        self.assertTrue(has_build_info)
32        self.assertEqual(record["base_ref"], "v0.210.0")
33
34
35if __name__ == "__main__":
36    unittest.main()
37