Add more tests for tree_parser