statement-to-tikz / tests /test_emit.py
will702's picture
Upload folder using huggingface_hub
eab734a verified
Raw
History Blame Contribute Delete
885 Bytes
"""TikZ emitter tests."""
from __future__ import annotations
from helpers import load_fixture
from statement_to_tikz.emit_tikz import emit_standalone_tex, emit_tikzpicture
from statement_to_tikz.solve import solve_geometry
def test_emit_contains_coordinates_and_segments() -> None:
ir, _ = load_fixture("equilateral.json")
scene = solve_geometry(ir)
body = emit_tikzpicture(scene)
assert r"\begin{tikzpicture}" in body
assert r"\coordinate (A)" in body
assert r"\draw (A) -- (B);" in body
assert "A" in body and "B" in body and "C" in body
def test_standalone_has_document_class() -> None:
ir, _ = load_fixture("right_altitude.json")
scene = solve_geometry(ir)
tex = emit_standalone_tex(scene)
assert r"\documentclass[11pt]{article}" in tex
assert r"\usetikzlibrary{calc,angles,quotes}" in tex
assert r"\end{document}" in tex