code: def _test_basic_flow(self, return_type):
      x = np.random.random((100, 16))
      y = np.random.random((100, 4))
      sample_weight = np.random.random((100,))
      batch_size = 16
      shuffle = True
      iterator = epoch_iterator.EpochIterator(
          x=x,
          y=y,
          sample_weight=sample_weight,
          batch_size=batch_size,
          shuffle=shuffle,
      )
      steps_seen = []
      for step, batch in iterator.enumerate_epoch(
          return_type=return_type
      ):
          batch = batch[0]
          steps_seen.append(step)
          self.assertEqual(len(batch), 3)
          if return_type == "np":
              self.assertIsInstance(batch[0], np.ndarray)
          else:
              self.assertIsInstance(batch[0], tf.Tensor)
      self.assertEqual(steps_seen, [0, 1, 2, 3, 4, 5, 6])
  
imports: import numpy
  import pytest
  import tensorflow
  from keras import backend
  from keras import testing
  from keras.trainers import data_adapters
  from keras.trainers import epoch_iterator
label_desc: Use image node_id 1 for calling the TestEpochIterator obj's underlying member method code with example usage: obj._test_basic_flow(return_type) without return types