Update unit test for tree algorithm.