#!/usr/bin/python
from lxml import etree
from subprocess import Popen, PIPE, STDOUT
import unittest
class OutputTests(unittest.TestCase):
def setUp(self):
print("")
def validateXMLTree(self, input, expected):
p = Popen(['../Build/blahtex', '--mathml', '--spacing', 'moderate'], stdout=PIPE, stdin=PIPE, stderr=STDOUT)
p.stdin.write(input)
p.stdin.close()
p.wait()
parser = etree.XMLParser(remove_blank_text=True)
actual = p.stdout.read()
rootNode = etree.XML(actual, parser=parser)
expectedRootNode = etree.XML(expected, parser=parser)
o1 = etree.tostring(rootNode)
o2 = etree.tostring(expectedRootNode)
self.assertEquals(o1, o2)
class TextTests(OutputTests):
def testRawText(self):
output = """
test
"""
self.validateXMLTree("test", output)
def testRawText2(self):
output = """
compute(T)
"""
self.validateXMLTree("compute(T)", output)
class ExpressionTests(OutputTests):
def testExpr(self):
output = """
n=1
"""
self.validateXMLTree("n = 1", output)
def testExpr2(self):
output = """
n=1,...,k-1
"""
self.validateXMLTree("n = 1, ... , k - 1", output)
def testExpr3(self):
output = """
n≠3
"""
self.validateXMLTree("n \\neq 3", output)
def testExpr4(self):
output = """
≤m
"""
self.validateXMLTree("\\leq m", output)
def testExpr5(self):
output = """
xy=st
"""
self.validateXMLTree("xy = st", output)
class EquationTests(OutputTests):
def testEq(self):
output = """
n=2h+1-1
"""
self.validateXMLTree("n = 2^{h+1} - 1", output)
def testEq2(self):
output = """
θ(n)=θ(2h+1-1)
"""
self.validateXMLTree("\\theta(n) = \\theta(2^{h+1} - 1)", output)
def testEq3(self):
output = """
S=G-{e}
"""
self.validateXMLTree("S = G - \\{ e \\}", output)
def testEq4(self):
output = """
λS=3k,k∈Z+
"""
self.validateXMLTree("\\lambda_S = 3k, k \\in \\mathbb{Z}^+", output)
self.validateXMLTree("\\lambda_S = 3k, k \\in \\mathbb{Z}^{+}", output)
def testEq5(self):
output = """
λS2=3b
"""
self.validateXMLTree("\\lambda_{S2} = 3b", output)
def testEqLong(self):
output = """
2(k-1)-1+2=2k-2-1+2=2k-1
"""
self.validateXMLTree("2(k - 1) - 1 + 2 = 2k - 2 - 1 + 2 = 2k - 1", output)
class SymbolTests(OutputTests):
def testTheta(self):
output = """
θ(n)
"""
self.validateXMLTree("\\theta(n)", output)
def testTheta2(self):
output = """
θ(2h)
"""
self.validateXMLTree("\\theta(2^h)", output)
def testFact(self):
output = """
10!-n=11*b
"""
self.validateXMLTree("10! - n = 11*b", output)
def testCup(self):
output = """
|A∪B∪C|=|A|+|B|+|C|-|A∩B|-|B∩C|-|A∩C|+|A∩B∩C|
"""
self.validateXMLTree("| A \cup B \cup C | = | A | + | B | + | C | - | A \cap B | - | B \cap C | - | A \cap C | + | A \cap B \cap C |", output)
class VariableTests(OutputTests):
def testVariables(self):
output = """
x,y,s,t,m,n∈R
"""
self.validateXMLTree("x, y, s, t, m, n \\in \\mathbb{R}", output)
def testVariables2(self):
output = """
f:N→N
"""
self.validateXMLTree("f : \\mathbb{N} \\rightarrow \\mathbb{N}", output)
def testVariables3(self):
output = """
g:N2→Z
"""
self.validateXMLTree("g : \\mathbb{N}^2 \\rightarrow \\mathbb{Z}", output)
class FunctionTests(OutputTests):
def testFunction(self):
output = """
g(m,n)=(2-n)f(m)
"""
self.validateXMLTree("g(m,n) = (2 - n) f(m)", output)
class ScriptTests(OutputTests):
def testSubscript(self):
output = """
W3
"""
self.validateXMLTree("W_3", output)
if __name__ == '__main__':
unittest.main()