Unit tests for AST functions

[luismeyer]

There should be unit tests for functions that interact with the AST. Those are located in the 'src/ast.ts'.