def METHOD_NAME(self):
    train_batch_sampler = SamplerHelper(self.train_ds)
    shuffle_sampler = train_batch_sampler.shuffle(buffer_size=10, seed=102)
    expected_result = {0: 4, 1: 9}
    for i, sample in enumerate(shuffle_sampler):
        if i in expected_result.keys():
            self.check_output_equal(sample, expected_result[i])