@Test
public void getFirstInCellTest() {

    int[][] cellData = new int[][]{{0, 1}, {2, 3, 4}, {5, 6}};
    Partition p = new Partition(cellData);
    for (int i = 0; i < cellData.length; i++) {
        Assert.assertEquals(cellData[i][0], p.getFirstInCell(i));
    }
}