def test_add_class_with_field_ctor_and_method(): parser = Parser("test.hpp") parser.init_ast() parser.add_class("MyClass") assert_equal(len(parser.ast.nodes), 1) parser.add_ctor() assert_equal(len(parser.ast.nodes[0].nodes), 1) parser.add_param("a", "int") assert_equal(len(parser.ast.nodes[0].nodes[0].nodes), 1) parser.add_method("myMethod", "int") assert_equal(len(parser.ast.nodes[0].nodes), 2) parser.add_field("b", "bool") assert_equal(len(parser.ast.nodes[0].nodes), 3)
def test_add_template_method(): parser = Parser("test.hpp") parser.init_ast() parser.add_class("MyClass") assert_equal(len(parser.ast.nodes), 1) parser.add_ctor() assert_equal(len(parser.ast.nodes[0].nodes), 1) parser.add_param("a", "int") assert_equal(len(parser.ast.nodes[0].nodes[0].nodes), 1) parser.add_template_method("myMethod", "T") assert_equal(len(parser.ast.nodes[0].nodes), 2) parser.add_template_type("T") assert_equal(len(parser.ast.nodes[0].nodes[1].template_types[0]), 1) parser.add_field("b", "T") assert_equal(len(parser.ast.nodes[0].nodes), 3)