def METHOD_NAME(self):
    splitter = TableDatasetSplitter(
        dataset_name="test",
        dataset_size=10000000,
        shard_size=100,
        num_epochs=1,
    )
    splitter.create_shards()
    shards = splitter.get_shards()
    self.assertEqual(len(shards), 50000)
    self.assertEqual(shards[-1].start, 4999900)
    self.assertEqual(shards[-1].end, 5000000)
    self.assertEqual(shards[0].name, "test")
    self.assertEqual(splitter.epoch, 1)
    splitter.create_shards()
    shards = splitter.get_shards()
    self.assertEqual(shards[0].start, 5000000)
    self.assertEqual(shards[0].end, 5000100)
    self.assertEqual(splitter.epoch, 2)