Files
buddha-gpt/tests/test_corpus.py
2026-08-14 17:06:26 -07:00

18 lines
669 B
Python

import json
from pathlib import Path
from buddhagpt.corpus import segments_to_text, load_bilara_suttas
def test_segments_to_text_joins_in_key_order():
segs = {"mn21:1.2": "second.", "mn21:1.1": "First."}
assert segments_to_text(segs) == "First. second."
def test_load_bilara_suttas(tmp_path):
d = tmp_path / "translation/en/sujato/sutta/mn"
d.mkdir(parents=True)
(d / "mn21_translation-en-sujato.json").write_text(
json.dumps({"mn21:0.1": "Middle Discourses 21", "mn21:1.1": "So I have heard."})
)
suttas = list(load_bilara_suttas(tmp_path))
assert suttas[0]["uid"] == "mn21"
assert "So I have heard." in suttas[0]["text"]