def test_puts_function(self): self.assertEqual( self.parse(""" int puts(const char * string) { int i = 0; while (string[i] != NULL) { putc(string[i++]); } putc('\\n'); return i + 1; } """), Program([ Function( return_type=Type(base='int'), name="puts", params=[ VariableDeclaration(name="string", vtype=Type( base="pointer", const=True, reference=Type(base="char"))) ], body=Block([ VariableDeclaration(name="i", vtype=Type(base="int"), value=Int32(value=0)), For(condition=BinaryOperation( operator="!=", left=ArrayDereference( array=Variable(name="string"), index=Variable(name="i")), right=Null()), body=Block([ Call(name="putc", args=[ ArrayDereference( array=Variable("string"), index=PostOperation( operator="++", variable=Variable(name="i"))) ]) ])), Call(name="putc", args=[Char('\n')]), ReturnStatement( value=BinaryOperation(operator="+", left=Variable(name="i"), right=Int32(value=1))) ])) ]))
def test_for_loop(self): self.assertEqual( self.parse( self.function_wrap("for ( i = 0; i < 10; i++ ) { putc(i); }")), self.function_wrap_node( For( initial=Assignment(left=Variable(name="i"), right=Int32(value=0)), condition=BinaryOperation(operator="<", left=Variable(name="i"), right=Int32(value=10)), increment=PostOperation(operator="++", variable=Variable(name="i")), body=Block([Call(name="putc", args=[Variable(name="i")])]), )))
def test_main_function(self): self.assertEqual( self.parse(""" int main(void) { return puts("Hello, world!"); } """), Program([ Function(return_type=Type(base='int'), name="main", params=[], body=Block([ ReturnStatement(value=Call( name="puts", args=[String("Hello, world!")])) ])) ]))
def test_while_loop(self): self.assertEqual( self.parse( self.function_wrap( "while (string[i] != NULL) { putc(string[i++]); }")), self.function_wrap_node( For(condition=BinaryOperation( operator="!=", left=ArrayDereference(array=Variable(name="string"), index=Variable("i")), right=Null()), body=Block([ Call(name="putc", args=[ ArrayDereference( array=Variable("string"), index=PostOperation( operator="++", variable=Variable(name="i"))) ]) ]))))
def test_auxiliary_function(self): self.assertEqual( self.parse(""" int foo(void) { return 12; } int main(void) { return foo(); } """), Program([ Function(return_type=Type(base='int'), name="foo", params=[], body=Block( [ReturnStatement(value=Int32(value=12))], )), Function(return_type=Type(base='int'), name="main", params=[], body=Block([ ReturnStatement(value=Call(name="foo", args=[]), ), ])), ]))
def function_call(self, p): return Call(name=p[0].getstr(), args=p[2].get_items())
def test_function_call_without_arguments(self): self.assertEqual( self.parse(self.function_wrap("putc();")), self.function_wrap_node(Call(name="putc", args=[])), )
def test_function_call_with_arguments(self): self.assertEqual( self.parse(self.function_wrap("putc(string);")), self.function_wrap_node( Call(name="putc", args=[Variable(name="string")])))