Extend StatementVisitor test