def METHOD_NAME(self):
    splitter = TextDatasetSplitter(
        dataset_name="test",
        dataset_size=1000,
        shard_size=10,
        num_epochs=1,
        shuffle=True,
    )
    splitter.create_shards()
    shards = splitter.get_shards()
    self.assertEqual(len(shards), 100)
    self.assertEqual(shards[0].start, 0)
    self.assertEqual(shards[0].end, 10)
    self.assertNotEqual(shards[0].record_indices, list(range(10)))
    self.assertEqual(shards[0].name, "test")
    self.assertEqual(splitter.epoch, 1)