class_docstr: Integration tests for clustering MHA layer.
code: def setUp(self):
      super(ClusterMHAIntegrationTest, self).setUp()
      self.x_train = np.random.uniform(size=(500, 32, 32))
      self.y_train = np.random.randint(low=0, high=1024, size=(500,))
  
      self.nr_of_clusters = 16
      self.params_clustering = {
          "number_of_clusters": self.nr_of_clusters,
          "cluster_centroids_init": CentroidInitialization.KMEANS_PLUS_PLUS,
      }
  
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 1 for calling the ClusterMHAIntegrationTest obj's underlying member method code with example usage: obj.setUp() without return types