Update test_treelib.py