def test_div(): src = ''' int main() { int i = 4500; int j = -123; printd(i/j); }''' assert [('FUNC_DEF', ('TYPE', 'int'), (('ID', 'main'), None), ('COMP_STATS', [('VAR_DEC', ('TYPE', 'int'), [('INIT_ASSIGN', ('ID', 'i'), ('ICONST', '4500'))]), ('VAR_DEC', ('TYPE', 'int'), [('INIT_ASSIGN', ('ID', 'j'), ('NEG', ('ICONST', '123')))]), ('STAT', ('FUNC_CALL', ('ID', 'printd'), [('DIV', ('ID', 'i'), ('ID', 'j'))]))]))] == \ parser.parse(src)
def test_var_dec(): assert [('VAR_DEC', ('TYPE', 'int'), [ ('INIT_ASSIGN', ('ID', 'sum'), ('ICONST', '1')) ])] == parser.parse("int sum = 1;")
def test_var_dec(): assert [('VAR_DEC', ('TYPE', 'int'), [('INIT_ASSIGN', ('ID', 'sum'), ('ICONST', '1'))] )] == parser.parse("int sum = 1;")