from anki.template2 import _removeFormattingFromMathjax


def test_remove_formatting_from_mathjax():
    assert _removeFormattingFromMathjax(r"\(2^{{c3::2}}\)", 3) == r"\(2^{{C3::2}}\)"

    txt = (
        r"{{c1::ok}} \(2^2\) {{c2::not ok}} \(2^{{c3::2}}\) \(x^3\) "
        r"{{c4::blah}} {{c5::text with \(x^2\) jax}}"
    )
    # Cloze 2 is not in MathJax, so it should not get protected against
    # formatting.
    assert _removeFormattingFromMathjax(txt, 2) == txt

    txt = r"\(a\) {{c1::b}} \[ {{c1::c}} \]"
    assert _removeFormattingFromMathjax(txt, 1) == (r"\(a\) {{c1::b}} \[ {{C1::c}} \]")