code: def test_compute_retrieval_metric(
      self,
      y_true,
      y_pred,
      input_type,
      metrics,
      k_list,
      expected,
  ):
      confusion_matrix = _retrieval.TopKRetrievalAggFn(
          metrics=metrics,
          k_list=k_list,
          input_type=input_type,
      )
      np.testing.assert_allclose(
          expected, confusion_matrix(y_true, y_pred)
      )
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  from tensorflow_model_analysis.experimental.metrics.aggregates import _retrieval
label_desc: Use image node_id 1 for calling the ClassificationTest obj's underlying member method code with example usage: obj.test_compute_retrieval_metric(y_true, y_pred, input_type, metrics, k_list, expected) without return types