[FEAT] add Tree edge case tests