49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
"""P2：一份起点/番茄导出 → 一份归因报告。"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import pathlib
|
||
import sys
|
||
import tempfile
|
||
import unittest
|
||
|
||
ROOT = next(
|
||
parent
|
||
for parent in (pathlib.Path(__file__).resolve().parent, *pathlib.Path(__file__).resolve().parents)
|
||
if (parent / "AGENTS.md").is_file() and (parent / ".git").exists()
|
||
)
|
||
if str(ROOT) not in sys.path:
|
||
sys.path.insert(0, str(ROOT))
|
||
|
||
from muse.feedback import import_ranking, write_attribution_report # noqa: E402
|
||
from muse.store import connect # noqa: E402
|
||
|
||
|
||
class RankingAttributionTest(unittest.TestCase):
|
||
def test_import_produces_non_empty_report(self) -> None:
|
||
tmp = pathlib.Path(tempfile.mkdtemp())
|
||
db = tmp / "muse.db"
|
||
with connect(db) as conn:
|
||
conn.execute(
|
||
"""INSERT INTO runs(id, created_at, kind, input_json, output_text, meta_json, skill_set_hash)
|
||
VALUES ('run-rank', datetime('now'), 'agent.writer', '{}', '稿', '{}', 'skills-abc')"""
|
||
)
|
||
conn.commit()
|
||
csv_path = tmp / "qidian.csv"
|
||
csv_path.write_text(
|
||
"platform,title,rank,votes,date\n起点,深空之影,8,12000,2026-08-01\n番茄,深空之影,3,88000,2026-08-01\n",
|
||
encoding="utf-8",
|
||
)
|
||
report = import_ranking(csv_path, sqlite_path=db)
|
||
self.assertEqual(report["imported"], 2)
|
||
dest = write_attribution_report(report, tmp / "attribution.md")
|
||
text = dest.read_text(encoding="utf-8")
|
||
self.assertIn("深空之影", text)
|
||
self.assertIn("run-rank", text)
|
||
self.assertIn("skills-abc", text)
|
||
self.assertGreater(len(text.strip()), 20)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|