Added TreeWalker tests
Address review comments
Merge pull request #1235 from ChrisParis/TreeWalker-tests

Added TreeWalker tests