refactor: avoid parsing document twice for figure numbers

test/test_photo.py

@@ -1,19 +1,17 @@
-import mistletoe
 import pytest
 
+from renderer.document import Document
 from renderer.fifty_ohm_html_renderer import FiftyOhmHtmlRenderer
 from renderer.fifty_ohm_latex_renderer import FiftyOhmLaTeXRenderer
-from test.util import paragraph
 
 
 @pytest.mark.html
 def test_photo_html():
-    assertions = {
-        "[photo:123:abc:Text]": FiftyOhmHtmlRenderer.render_photo_helper("123", "abc", "Text", "TODO", ""),
-    }
+    assertions = {"[photo:123:abc:Text]": ("123", "abc", "Text", "A-5.7.1", "")}
 
-    for assertion in assertions:
-        assert mistletoe.markdown(assertion, FiftyOhmHtmlRenderer) == paragraph(assertions[assertion])
+    with FiftyOhmHtmlRenderer(edition="A", chapter="5", section="7") as renderer:
+        for assertion in assertions:
+            assert renderer.render(Document(assertion)) == renderer.render_photo_helper(*assertions[assertion]) + "\n"
 
 
 @pytest.mark.latex
@@ -22,5 +20,6 @@ def test_photo_latex():
         "[photo:123:abc:Text]": FiftyOhmLaTeXRenderer.render_photo_helper("123", "abc", "Text", "TODO"),
     }
 
-    for assertion in assertions:
-        assert mistletoe.markdown(assertion, FiftyOhmLaTeXRenderer) == "\n" + assertions[assertion] + "\n"
+    with FiftyOhmLaTeXRenderer() as renderer:
+        for assertion in assertions:
+            assert renderer.render(Document(assertion)) == assertions[assertion]