code: def _create_test_data(self, model):
      x_train = np.random.randn(
          *self._batch(model.input.get_shape().as_list(), 2)
      ).astype("float32")
      y_train = tf.keras.utils.to_categorical(
          np.random.randint(1000, size=(2, 1)), 1000
      )
  
      return x_train, y_train
  
file_path: model-optimization/tensorflow_model_optimization/python/core/quantization/keras/quantize_models_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import inspect
  import tempfile
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.quantization.keras import quantize
  from tensorflow_model_optimization.python.core.quantization.keras import utils
label_desc: Use image node_id 3 for calling the QuantizeModelsTest obj's underlying member method code with example usage: obj._create_test_data(model) and returns: x_train,  y_train