Update TestPartition.java