def test_modify_mfenced():
modify_mml = '''
<mtd>
<mn>2</mn>
<mfenced>
<mrow>
<mi>x</mi>
<mo>-</mo>
<mn>4</mn>
</mrow>
</mfenced>
</mtd>
'''
modify_to_mml = flatten_string('''
<mtd>
<mmul>
<mn> 2 </mn>
<mfenced>
<mrow>
<mi> x </mi>
<mo> - </mo>
<mn> 4 </mn>
</mrow>
</mfenced>
</mmul>
</mtd>
''')
tree = mml2tree(modify_mml)
modified_tree = modify(tree)
assert etree.tostring(modified_tree).decode('utf-8') == modify_to_mml
def test_modify__highest_priority_ops():
modify_mml = '''
<mtd>
<mn> 2 </mn>
<mo> × </mo>
<mo> - </mo>
<mn> 4 </mn>
</mtd>
'''
tree = mml2tree(modify_mml)
ops = _highest_priority_ops(tree.getchildren())
assert len(ops) == 1
assert ops[0].text.strip() == '×'
def test_modify_positive_negative_first_element():
modify_mml = '''
<mtd>
<mo> + </mo>
<mi> x </mi>
</mtd>
'''
modify_to_mml = b'<mtd><mi> x </mi></mtd>'
tree = mml2tree(modify_mml)
modified_tree = modify(tree)
assert etree.tostring(modified_tree) == modify_to_mml
modify_mml = '''
<mtd>
<mo> - </mo>
<mi> x </mi>
</mtd>
'''
modify_to_mml = b'<mtd><mmul><mn> -1 </mn><mi> x </mi></mmul></mtd>'
tree = mml2tree(modify_mml)
modified_tree = modify(tree)
assert etree.tostring(modified_tree) == modify_to_mml
def test_modify_times():
modify_mml = '''
<mtd>
<mn> 2 </mn>
<mo> × </mo>
<mo> - </mo>
<mn> 4 </mn>
</mtd>
'''
modify_to_mml = '''
<mtd>
<mmul>
<mn> 2 </mn>
<mn> -4 </mn>
</mmul>
</mtd>
'''
tree = mml2tree(modify_mml)
modified_tree = modify(tree)
expected_tree = mml2tree(modify_to_mml)
assert modified_tree.tag == expected_tree.tag
assert hasattr(modified_tree, 'mmul')
assert hasattr(modified_tree.mmul, 'mn')
assert modified_tree.mmul.countchildren() == 2
def test_tree2sympy():
tree_mml = '''
<meq>
<madd>
<mmul>
<mn>2</mn>
<mi>x</mi>
</mmul>
<mn>4</mn>
</madd>
<mn>7</mn>
</meq>
'''
tree = mml2tree(tree_mml)
sympy = tree2sympy(tree)
assert sympy == "Eq(Add(Mul(Integer(2),Symbol('x'),evaluate=False),Integer(4),evaluate=False),Integer(7),evaluate=False)"
tree_mml = '''
<mfrac>
<mrow>
<madd>
<mmul>
<mn>2</mn>
<mi>x</mi>
</mmul>
<mn>4</mn>
</madd>
</mrow>
<mrow>
<mn>7</mn>
</mrow>
</mfrac>
'''
tree = mml2tree(tree_mml)
sympy = tree2sympy(tree)
assert sympy == "Mul(Add(Mul(Integer(2),Symbol('x'),evaluate=False),Integer(4),evaluate=False),Pow(Integer(7),Integer(-1)),evaluate=False)"
def test_table2trees():
mtable_mml = '''
<mtable columnalign="left">
<mtr>
<mtd>
<mn> 2 </mn>
<mi> x </mi>
<mo> - </mo>
<mn> 4 </mn>
<mo> = </mo>
<mn> 7 </mn>
</mtd>
</mtr>
<mtr>
<mtd>
<mn> 2 </mn>
<mi> x </mi>
<mo> = </mo>
<mn> 11 </mn>
</mtd>
</mtr>
<mtr>
<mtd>
<mi> x </mi>
<mo> = </mo>
<mfrac>
<mrow>
<mn> 11 </mn>
</mrow>
<mrow>
<mn> 2 </mn>
</mrow>
</mfrac>
</mtd>
</mtr>
</mtable>
'''
trees = table2trees(mml2tree(mtable_mml))
assert isinstance(trees, list)
assert len(trees) == 3
assert len(trees[0].getchildren()) == 6
assert len(trees[1].getchildren()) == 4
assert len(trees[2].getchildren()) == 3
def test_modify():
modify_mml = '''
<mtd>
<mn> 2 </mn>
<mi> x </mi>
<mo> - </mo>
<mn> 4 </mn>
<mo> + </mo>
<mn> 7 </mn>
</mtd>
'''
modify_to_mml = '''
<mtd>
<madd>
<mmul>
<mn>2</mn>
<mi>x</mi>
</mmul>
<mn>4</mn>
<mn>7</mn>
</madd>
</mtd>
'''
tree = mml2tree(modify_mml)
modified_tree = modify(tree)
modify_tree = mml2tree(modify_to_mml)
assert modify_tree.tag == modified_tree.tag
print(etree.tostring(modified_tree))
assert hasattr(modified_tree, 'madd')
assert hasattr(modified_tree.madd, 'mn')
assert hasattr(modified_tree.madd, 'mmul')
assert hasattr(modified_tree.madd.mmul, 'mi')
modify_mml = '''
<mtd>
<mn> 2 </mn>
<mi> x </mi>
<mo> - </mo>
<mn> 4 </mn>
<mo> = </mo>
<mn> 7 </mn>
</mtd>
'''
modify_to_mml = '''
<mtd>
<meq>
<madd>
<mmul>
<mn>2</mn>
<mi>x</mi>
</mmul>
<mn>4</mn>
</madd>
<mn>7</mn>
</meq>
</mtd>
'''
tree = mml2tree(modify_mml)
modified_tree = modify(tree)
modify_tree = mml2tree(modify_to_mml)
assert modify_tree.tag == modified_tree.tag
assert hasattr(modified_tree, 'meq')
assert modified_tree.meq.countchildren() == 2
assert hasattr(modified_tree.meq, 'madd')
assert hasattr(modified_tree.meq.madd, 'mmul')
def test_mml2tree():
mml = '''
<math xmlns="http://www.w3.org/1998/Math/MathML">
<mstyle displaystyle="true">
<mn> 2 </mn>
<mi> x </mi>
<mo> - </mo>
<mn> 4 </mn>
<mo> = </mo>
<mn> 7 </mn>
</mstyle>
</math>
'''
tree = mml2tree(mml)
assert tree is not None
assert tree.tag == 'math'
assert hasattr(tree, 'mstyle')
mml = '''
<math xmlns="http://www.w3.org/1998/Math/MathML">
<mstyle displaystyle="true">
<mtable columnalign="left">
<mtr>
<mtd>
<mn> 2 </mn>
<mi> x </mi>
<mo> - </mo>
<mn> 4 </mn>
<mo> = </mo>
<mn> 7 </mn>
</mtd>
</mtr>
<mtr>
<mtd>
<mn> 2 </mn>
<mi> x </mi>
<mo> = </mo>
<mn> 11 </mn>
</mtd>
</mtr>
<mtr>
<mtd>
<mi> x </mi>
<mo> = </mo>
<mfrac>
<mrow>
<mn> 11 </mn>
</mrow>
<mrow>
<mn> 2 </mn>
</mrow>
</mfrac>
</mtd>
</mtr>
</mtable>
</mstyle>
</math>
'''
tree = mml2tree(mml)
assert tree is not None
assert tree.tag == 'math'
assert hasattr(tree, 'mstyle')
assert hasattr(tree.mstyle, 'mtable')
mstyle = tree.xpath('/math/mstyle')[0]
assert mstyle is not None
assert hasattr(mstyle, 'mtable')