KeyDistribution keyDistri = reader.getKeyDistribution(numSplits * 10, 1, lastBd);
Assert.assertEquals(totalBytes, keyDistri.length()+lastBd.getLength());
reader.close();
BytesWritable[] keys = null;
if (keyDistri.size() >= numSplits) {
keyDistri.resize(lastBd);
Assert.assertEquals(totalBytes, keyDistri.length()+lastBd.getLength());
RawComparable[] rawComparables = keyDistri.getKeys();
keys = new BytesWritable[rawComparables.length];
for (int i = 0; i < keys.length; ++i) {
keys[i] = new BytesWritable();