def test_subroutine_call(self):
src = '\n'.join([
'<expression>',
'<term>',
'<identifier> SquareGame </identifier>',
'<symbol> . </symbol>',
'<identifier> new </identifier>',
'<symbol> ( </symbol>',
'<expressionList>',
'<expression>',
'<term>',
'<integerConstant> 0 </integerConstant>',
'</term>',
'</expression>',
'</expressionList>',
'<symbol> ) </symbol>',
'</term>',
'</expression>',
])
et = ET.fromstring(src)
node = cp.el_to_node(et)
self.generator.generate(node)
self.assertEqual(
self.stream.getvalue(), '\n'.join([
'push constant 0',
'call SquareGame.new 1\n',
]))
def test_let(self):
src = '\n'.join([
'<letStatement>',
'<keyword> let </keyword>',
'<identifier> s </identifier>',
'<symbol> = </symbol>',
'<expression>',
'<term>',
'<keyword> null </keyword>',
'</term>',
'</expression>',
'<symbol> ; </symbol>',
'</letStatement>',
])
el = ET.fromstring(src)
node = cp.el_to_node(el)
self.generator.sym_tab.mth = {
's': {
'kind': 'static',
'number': '2',
}
}
self.generator.generate(node)
self.assertEqual(self.stream.getvalue(), '\n'.join([
'push constant 0',
'pop static 2\n',
]))
def test_op(self):
src = '<symbol> * </symbol>'
el = ET.fromstring(src)
node = cp.el_to_node(el)
self.generator.generate(node)
self.assertEqual(
self.stream.getvalue(),
'mult\n',
)
def test_term(self):
src = '\n'.join([
'<term>',
'<integerConstant> 5 </integerConstant>',
'</term>',
])
el = ET.fromstring(src)
node = cp.el_to_node(el)
self.generator.generate(node)
self.assertEqual(
self.stream.getvalue(),
'push constant 5\n',
)
def test_string(self):
src = '<stringConstant> fo </stringConstant>'
el = ET.fromstring(src)
node = cp.el_to_node(el)
self.generator.generate(node)
self.assertEqual(
self.stream.getvalue(), '\n'.join([
'push constant 2',
'String.new 1',
'push constant 102',
'String.appendChar 2',
'push constant 111',
'String.appendChar 2\n',
]))
def test_expr(self):
src = '\n'.join([
'<expression>',
'<term>',
'<integerConstant> 8 </integerConstant>',
'</term>',
'<symbol> * </symbol>',
'<term>',
'<integerConstant> 5 </integerConstant>',
'</term>',
'</expression>',
])
el = ET.fromstring(src)
node = cp.el_to_node(el)
self.generator.generate(node)
self.assertEqual(
self.stream.getvalue(), '\n'.join([
'push constant 8',
'push constant 5',
'mult\n',
]))
def test_let_array(self):
src = '\n'.join([
'<letStatement>',
'<keyword> let </keyword>',
'<identifier> s </identifier>',
'<symbol> [ </symbol>',
'<expression>',
'<term>',
'<integerConstant> 19 </integerConstant>',
'</term>',
'</expression>',
'<symbol> = </symbol>',
'<expression>',
'<term>',
'<keyword> null </keyword>',
'</term>',
'</expression>',
'<symbol> ; </symbol>',
'</letStatement>',
])
el = ET.fromstring(src)
node = cp.el_to_node(el)
self.generator.sym_tab.mth = {
's': {
'kind': 'local',
'number': '2',
}
}
self.generator.generate(node)
self.assertEqual(
self.stream.getvalue(), '\n'.join([
'push local 2',
'push constant 19',
'add',
'pop pointer 1',
'push constant 0',
'pop that 0\n',
]))
def test_integer(self):
src = '<integerConstant> 5 </integerConstant>'
element = ET.fromstring(src)
node = cp.el_to_node(element)
self.generator.generate(node)
self.assertEqual(self.stream.getvalue(), 'push constant 5\n')
def setUp(self):
super(TestSubroutineDec, self).setUp()
self.class_var_decs = [e for e in self.el if e.tag == 'classVarDec']
for cvd in self.class_var_decs:
self.generator.generate(cp.el_to_node(cvd))
self.el = next(e for e in self.el if e.tag == 'subroutineDec')
def node(self):
return cp.el_to_node(self.el)