def METHOD_NAME(self):
    train_batch_sampler = SamplerHelper(self.train_ds)
    self.check_output_equal(len(train_batch_sampler), 10)
    self.check_output_equal(len(train_batch_sampler), train_batch_sampler.length)
    train_batch_sampler.length = 5
    self.check_output_equal(len(train_batch_sampler), 5)