class_docstr: Integration tests for clustering.
code: def dataset_generator(self):
      for x, y in zip(self.x_train, self.y_train):
          yield np.array([x]), np.array([y])
  
imports: import os
  import tempfile
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.clustering.keras import cluster
  from tensorflow_model_optimization.python.core.clustering.keras import cluster_config
  from tensorflow_model_optimization.python.core.clustering.keras.experimental import cluster
label_desc: Use image node_id 2 for calling the ClusterIntegrationTest obj's underlying member method code with example usage: obj.dataset_generator() without return types