yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-09-25 16:30:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adding some tests

Browse Source
This commit is contained in:
Jake Poznanski 2025-03-11 03:57:12 +00:00
parent 35cc6f110c
commit bc3a94583a
1 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
olmocr/bench/katex/render.py
View File

@ -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()