@Test @Category(UnitTest.class) public void writeSplitsPath() throws IOException { Splits splits = new PartitionerSplit(); splits.generateSplits(new TestGenerator()); File splitfile = folder.newFolder(testName.getMethodName()); splits.writeSplits(new Path(splitfile.toURI())); FileInputStream in = new FileInputStream(new File(splitfile, "partitions")); Scanner reader = new Scanner(in); Assert.assertEquals("Wrong number written", generated.length, reader.nextInt()); PartitionerSplit.PartitionerSplitInfo[] si = (PartitionerSplit.PartitionerSplitInfo[]) splits.getSplits(); for (int i = 0; i < generated.length; i++) { Assert.assertEquals("Splits entry not correct", generated[i].longValue(), reader.nextLong()); Assert.assertEquals("Partition entry not correct", i, reader.nextLong()); } reader.close(); }