@Test
void testTableSplitting() {

    ImmutableTable<String, String, Double> tableWeights = ImmutableTableFactory.createWithRows(
            ImmutableMap.of(
                    "Column1", 1d,
                    "Column2", 2d),
            ImmutableList.of("Row1", "Row2"));
    ImmutableTable<String, String, Integer> splitInts = NonNegativeIntegerSplitterByWeights.splitByTableWeights(tableWeights, 2);
    Assertions.assertEquals(0, (int) splitInts.get("Row1", "Column1"));
    Assertions.assertEquals(1, (int) splitInts.get("Row1", "Column2"));
    Assertions.assertEquals(0, (int) splitInts.get("Row2", "Column1"));
    Assertions.assertEquals(1, (int) splitInts.get("Row2", "Column2"));
}