Map<HoodieFileGroupId, List<Integer>> getFileGroupToPartitions() {
    return fileGroupToPartitions;
  }
@Test
  public void testAssignmentCorrectness() {
    HoodieFileGroupId fg1 = new HoodieFileGroupId("p1", "f1");
    HoodieFileGroupId fg2 = new HoodieFileGroupId("p1", "f2");
    HoodieFileGroupId fg3 = new HoodieFileGroupId("p1", "f3");

    Map<HoodieFileGroupId, Long> fileToComparisons = new HashMap<HoodieFileGroupId, Long>() {
      {
        put(fg1, 40L);
        put(fg2, 35L);
        put(fg3, 20L);
      }
    };
    BucketizedBloomCheckPartitioner p = new BucketizedBloomCheckPartitioner(4, fileToComparisons, 10);
    Map<HoodieFileGroupId, List<Integer>> assignments = p.getFileGroupToPartitions();
    assertEquals(4, assignments.get(fg1).size(), "f1 should have 4 buckets");
    assertEquals(4, assignments.get(fg2).size(), "f2 should have 4 buckets");
    assertEquals(2, assignments.get(fg3).size(), "f3 should have 2 buckets");
    assertArrayEquals(new Integer[] {0, 0, 1, 3}, assignments.get(fg1).toArray(), "f1 spread across 3 partitions");
    assertArrayEquals(new Integer[] {2, 2, 3, 1}, assignments.get(fg2).toArray(), "f2 spread across 3 partitions");
    assertArrayEquals(new Integer[] {1, 0}, assignments.get(fg3).toArray(), "f3 spread across 2 partitions");
  }