Modified tests for TreeGraph structure