def METHOD_NAME(self):
    batch_size = 8
    steps_per_epoch = 6
    dataset_size = batch_size * (steps_per_epoch - 2)
    x = np.arange(dataset_size).reshape((dataset_size, 1))
    y = x * 2
    iterator = epoch_iterator.EpochIterator(
        x=x,
        y=y,
        batch_size=batch_size,
        steps_per_epoch=steps_per_epoch,
    )
    steps_seen = []
    for step, _ in iterator.enumerate_epoch():
        steps_seen.append(step)
    self.assertLen(steps_seen, steps_per_epoch - 2)
    self.assertIsInstance(iterator, epoch_iterator.EpochIterator)
    self.assertTrue(iterator._insufficient_data)