from anki.template import Template


def test_remove_formatting_from_mathjax():
    t = Template('')
    assert t._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 t._removeFormattingFromMathjax(txt, 2) == txt

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