add unit test for tree set