def test_type_visitor(self): cu = parse(TypeVisitor.java_source) tv = TypeVisitor() tv.visit(cu, None) assertTrue(ReferenceType in [type(n) for n in tv.getNodes()]) names = [str(node) for node in tv.getNodes()] assertTrue("VoidVisitorAdapter" in names) assertTrue("PyObject" in names) assertTrue("WildcardType" in names)
def test_annotation_visitor(self): cu = parse(TypeVisitor.java_source) av = AnnotationVisitor() av.visit(cu, None) assertTrue(str(av.getNodes()[0]) == "@Override")