Refactor TestTreePeek() to use checkTokens()