tests: TreeGeneratorTest updated