def add(self, pk: int, latex: str):
ast = self.__parse(latex)
visitor = IndexCreatorVisitor(self.data, pk)
ast.accept(visitor)
self.number_of_different_nodes[pk] = visitor.get_number_of_nodes_of_different_nodes()
def __init__(self, path: str = None):
self.data = {}
self.number_of_different_nodes = {}
if path:
with open(path, 'r') as f:
for line in f:
tokens = line.split(':')
pk = int(tokens[0])
latex = tokens[1].strip()
ast = self.__parse(latex)
visitor = IndexCreatorVisitor(self.data, pk)
ast.accept(visitor)
self.number_of_different_nodes[pk] = visitor.get_number_of_nodes_of_different_nodes()
def test_visit_basic_addition(self):
ast = self.parse(r'3 + x')
data = {}
pk = 1
visitor = IndexCreatorVisitor(data, pk)
ast.accept(visitor)
self.assertTrue(self.has(data[0][NodeType.CONSTANT], pk, '3'))
self.assertTrue(self.has(data[0][NodeType.VARIABLE], pk, 'x'))
self.assertTrue(pk in data[1][NodeType.ADDITION])
def test_visit_exponentiation_and_function(self):
ast = self.parse(r'e^{x!}')
data = {}
pk = 1
visitor = IndexCreatorVisitor(data, pk)
ast.accept(visitor)
self.assertTrue(self.has(data[0][NodeType.VARIABLE], pk, 'e'))
self.assertTrue(self.has(data[0][NodeType.VARIABLE], pk, 'x'))
self.assertTrue(pk in data[1][NodeType.FUNCTION])
self.assertTrue(pk in data[2][NodeType.EXPONENTIATION])
def test_visit_summation(self):
ast = self.parse(r'\sum_{i=0}^{\infty} i * i')
data = {}
pk = 1
visitor = IndexCreatorVisitor(data, pk)
ast.accept(visitor)
self.assertTrue(self.has(data[0][NodeType.CONSTANT], pk, '0'))
self.assertTrue(self.has(data[0][NodeType.BOUNDVARIABLE], pk, 'i'))
self.assertTrue(self.has(data[0][NodeType.CONSTANT], pk, '\\infty'))
self.assertTrue(pk in data[1][NodeType.MULTIPLICATION])
self.assertTrue(pk in data[2][NodeType.SUMMATION])
def test_visit_indexed_variable(self):
ast = self.parse(r'3*B_{2n}')
data = {}
pk = 1
visitor = IndexCreatorVisitor(data, pk)
ast.accept(visitor)
self.assertTrue(self.has(data[0][NodeType.CONSTANT], pk, '2'))
self.assertTrue(self.has(data[0][NodeType.CONSTANT], pk, '3'))
self.assertTrue(self.has(data[0][NodeType.VARIABLE], pk, 'n'))
self.assertTrue(pk in data[1][NodeType.MULTIPLICATION])
self.assertTrue(self.has(data[2][NodeType.INDEXEDVARIABLE], pk, 'B'))
self.assertTrue(pk in data[3][NodeType.MULTIPLICATION])
def test_visit_summation_with_free_and_bound_variable(self):
ast = self.parse(r'\sum_{i=0}^{\infty} (i + j)')
data = {}
pk = 1
visitor = IndexCreatorVisitor(data, pk)
ast.accept(visitor)
self.assertTrue(self.has(data[0][NodeType.CONSTANT], pk, '0'))
self.assertTrue(self.has(data[0][NodeType.BOUNDVARIABLE], pk, 'i'))
self.assertTrue(self.has(data[0][NodeType.VARIABLE], pk, 'j'))
self.assertTrue(self.has(data[0][NodeType.CONSTANT], pk, '\\infty'))
self.assertTrue(pk in data[1][NodeType.ADDITION])
self.assertTrue(pk in data[2][NodeType.SUMMATION])
def test_visit_addition_mixed_with_multiplication(self):
ast = self.parse(r'3 * x + y * 5')
data = {}
pk = 1
visitor = IndexCreatorVisitor(data, pk)
ast.accept(visitor)
self.assertTrue(self.has(data[0][NodeType.CONSTANT], pk, '3'))
self.assertTrue(self.has(data[0][NodeType.VARIABLE], pk, 'x'))
self.assertTrue(self.has(data[0][NodeType.VARIABLE], pk, 'y'))
self.assertTrue(self.has(data[0][NodeType.CONSTANT], pk, '5'))
self.assertTrue(pk in data[1][NodeType.MULTIPLICATION])
self.assertTrue(pk in data[2][NodeType.ADDITION])