class_docstr: Unit tests for clustering lookup algorithms.
class_docstr_tok: ["Unit","tests","for","clustering","lookup","algorithms","."]
code: def _check_pull_values(
      self, clustering_algo, pulling_indices, expected_output
  ):
      pulling_indices = tf.convert_to_tensor(pulling_indices)
  
      clustered_weight = clustering_algo.get_clustered_weight(
          pulling_indices,
          original_weight=tf.zeros(
              pulling_indices.shape, dtype=tf.float32
          ),
      )
      self.assertAllEqual(clustered_weight, expected_output)
  
imports: from absl.testing import parameterized
  import tensorflow
  from tensorflow_model_optimization.python.core.clustering.keras import clusterable_layer
  from tensorflow_model_optimization.python.core.clustering.keras import clustering_registry
  from tensorflow_model_optimization.python.core.clustering.keras.cluster_config import GradientAggregation
label_desc: Use image node_id 1 for calling the ClusteringAlgorithmTest obj's underlying member method code with example usage: obj._check_pull_values(clustering_algo, pulling_indices, expected_output) without return types