Update TreeTests.py