feat: add Quant OS A-share baseline

2026-07-26 12:54:04 +08:00
commit 48c5f64bbd
98 changed files with 31874 additions and 0 deletions
@@ -0,0 +1,46 @@
+import hashlib
+import json
+import tempfile
+import unittest
+from pathlib import Path
+
+from quant60.experiment import (
+    run_research_smoke,
+    verify_research_manifest,
+    write_research_artifacts,
+)
+
+
+class ResearchExperimentTests(unittest.TestCase):
+    def test_research_smoke_is_deterministic_and_verifiable(self):
+        first = run_research_smoke(trading_days=260, seed=11)
+        second = run_research_smoke(trading_days=260, seed=11)
+        self.assertEqual(first, second)
+        self.assertGreater(first["report"]["fold_count"], 0)
+        self.assertFalse(first["report"]["investment_value_claim"])
+        self.assertTrue(first["targets"])
+
+        with tempfile.TemporaryDirectory() as left_dir, tempfile.TemporaryDirectory() as right_dir:
+            left = write_research_artifacts(first, left_dir)
+            right = write_research_artifacts(second, right_dir)
+            for name in left:
+                self.assertEqual(
+                    hashlib.sha256(Path(left[name]).read_bytes()).hexdigest(),
+                    hashlib.sha256(Path(right[name]).read_bytes()).hexdigest(),
+                    name,
+                )
+            verified = verify_research_manifest(left["manifest"])
+            self.assertTrue(verified["ok"])
+
+    def test_research_verifier_rejects_tamper(self):
+        result = run_research_smoke(trading_days=260, seed=12)
+        with tempfile.TemporaryDirectory() as directory:
+            paths = write_research_artifacts(result, directory)
+            report = Path(paths["report"])
+            report.write_bytes(report.read_bytes() + b" ")
+            with self.assertRaisesRegex(ValueError, "hash mismatch"):
+                verify_research_manifest(paths["manifest"])
+
+
+if __name__ == "__main__":
+    unittest.main()