31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
from types import SimpleNamespace
|
|
|
|
from genesis.data_models import ParsedTemplate, ChapterMarker
|
|
from genesis.writer.context_builder import build_contexts
|
|
from genesis.writer.models import GenerationContext
|
|
|
|
|
|
def _ss():
|
|
template = ParsedTemplate(
|
|
file_name="t.docx",
|
|
sections=[
|
|
ChapterMarker(type="heading", name="はじめに", level=1),
|
|
ChapterMarker(type="placeholder", name="section:introduction", level=0),
|
|
],
|
|
placeholders={},
|
|
styles={"defined": ["Heading1"], "used": ["Heading1", "Normal"]},
|
|
)
|
|
return SimpleNamespace(template=template)
|
|
|
|
|
|
def test_build_contexts_maps_chapters():
|
|
ctxs = build_contexts(_ss(), samples_dir="nonexistent_dir_xyz")
|
|
assert len(ctxs) == 1
|
|
c = ctxs[0]
|
|
assert isinstance(c, GenerationContext)
|
|
assert c.chapter_id == "introduction"
|
|
assert c.write_rules == [] # 样本缺失 -> 空
|
|
assert c.design_rules == []
|
|
assert c.template_styles == {"Heading1", "Normal"}
|
|
assert c.template_marker.section_placeholder == "section:introduction"
|