47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
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()
|