@Test
public void getCellTest() {

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