Added assertion in treepartition method