From bc3a94583a8440f473e5a306e276d89bbf6f22d0 Mon Sep 17 00:00:00 2001 From: Jake Poznanski Date: Tue, 11 Mar 2025 03:57:12 +0000 Subject: [PATCH] Adding some tests --- olmocr/bench/katex/render.py | 21 ++++++++++++++++++--- 1 file changed, 18 insertions(+), 3 deletions(-) diff --git a/olmocr/bench/katex/render.py b/olmocr/bench/katex/render.py index 4708a6d..b8eeeb6 100644 --- a/olmocr/bench/katex/render.py +++ b/olmocr/bench/katex/render.py @@ -324,9 +324,9 @@ def compare_rendered_equations(haystack: RenderedEquation, needle: RenderedEquat haystack_inner = normalize(extract_inner(haystack.mathml)) needle_inner = normalize(extract_inner(needle.mathml)) - # # For debugging: print the cleaned MathML strings. - # print("Cleaned haystack MathML:", haystack_inner) - # print("Cleaned needle MathML:", needle_inner) + # For debugging: print the cleaned MathML strings. + print("Cleaned haystack MathML:", haystack_inner) + print("Cleaned needle MathML:", needle_inner) # If needle is longer than haystack, swap them. if len(needle_inner) > len(haystack_inner): @@ -370,5 +370,20 @@ class TestRenderedEquationComparison(unittest.TestCase): align_rendered = render_equation("""\\begin{align*}\\nabla \\cdot \\mathbf{E} = \\frac{\\rho}{\\varepsilon_0}\\end{align*}""", use_cache=False, debug_dom=False) self.assertTrue(compare_rendered_equations(ref_rendered, align_rendered)) + def test_dot_end1(self): + ref_rendered = render_equation("\\lambda_{g}=\\sum_{s \\in S} \\zeta_{n}^{\\psi(g s)}=\\sum_{i=1}^{k}\\left[\\sum_{s, R s=\\mathcal{I}_{i}} \\zeta_{n}^{\\varphi(g s)}\\right]") + align_rendered = render_equation("\\lambda_{g}=\\sum_{s \\in S} \\zeta_{n}^{\\psi(g s)}=\\sum_{i=1}^{k}\\left[\\sum_{s, R s=\\mathcal{I}_{i}} \\zeta_{n}^{\\varphi(g s)}\\right].") + self.assertTrue(compare_rendered_equations(ref_rendered, align_rendered)) + + def test_dot_end2(self): + ref_rendered = render_equation("\\lambda_{g}=\\sum_{s \\in S} \\zeta_{n}^{\\psi(g s)}=\\sum_{i=1}^{k}\\left[\\sum_{s, R s=\\mathcal{I}_{i}} \\zeta_{n}^{\\psi(g s)}\\right]") + align_rendered = render_equation("\\lambda_g = \\sum_{s \\in S} \\zeta_n^{\\psi(gs)} = \\sum_{i=1}^{k} \\left[ \\sum_{s, Rs = I_i} \\zeta_n^{\\psi(gs)} \\right]") + self.assertTrue(compare_rendered_equations(ref_rendered, align_rendered)) + + def test_lambda(self): + ref_rendered = render_equation("\\lambda_g = \\lambda_{g'}") + align_rendered = render_equation("\\lambda_{g}=\\lambda_{g^{\\prime}}") + self.assertTrue(compare_rendered_equations(ref_rendered, align_rendered)) + if __name__ == "__main__": unittest.main()