def test_dynamic_work_rebalancing_exhaustive(self):

    def compare_split_points(file_name):
        source = _create_avro_source(file_name, use_fastavro=self.use_fastavro)
        splits = [split for split in source.split(desired_bundle_size=float('inf'))]
        assert (len(splits) == 1)
        source_test_utils.assert_split_at_fraction_exhaustive(splits[0].source)
    file_name = self._write_data(count=5, sync_interval=2)
    compare_split_points(file_name)
def test_dynamic_work_rebalancing_exhaustive(self):

    def compare_split_points(file_name):
        source = _create_avro_source(file_name, use_fastavro=self.use_fastavro)
        splits = [split for split in source.split(desired_bundle_size=float('inf'))]
        assert (len(splits) == 1)
        source_test_utils.assert_split_at_fraction_exhaustive(splits[0].source)
    file_name = self._write_data(count=5, sync_interval=2)
    compare_split_points(file_name)