@Test
@Category(UnitTest.class)
public void readSplitsPath() throws IOException {
File splitFolder = folder.newFolder(testName.getMethodName());
File splitfile = new File(splitFolder, "partitions");
FileOutputStream stream = new FileOutputStream(splitfile);
PrintWriter writer = new PrintWriter(stream);
writer.println(generated.length);
for (int i = 0; i < generated.length; i++) {
writer.print(generated[i]);
writer.print(" ");
writer.println(i);
}
writer.close();
stream.close();
Splits splits = new PartitionerSplit();
splits.readSplits(new Path(splitFolder.toURI()));
PartitionerSplit.PartitionerSplitInfo[] si =
(PartitionerSplit.PartitionerSplitInfo[]) splits.getSplits();
Assert.assertEquals("Splits length not correct", generated.length, si.length);
for (int i = 0; i < generated.length; i++) {
Assert.assertEquals("Splits entry not correct", generated[i].longValue(), si[i].getTileId());
}
}
