code: def testInsertTwoPredictionLogsIntoExtracts(self):
      model_names_to_prediction_logs = {
          "prediction_log1": self.prediction_log1,
          "prediction_log2": self.prediction_log2,
      }
      inference_tuple = ({}, model_names_to_prediction_logs)
      extracts = inference_base.insert_predictions_into_extracts(
          inference_tuple,
          prediction_log_keypath=[constants.PREDICTION_LOG_KEY],
      )
  
      ref_extracts = {
          constants.PREDICTION_LOG_KEY: model_names_to_prediction_logs
      }
  
      self.assertEqual(
          extracts[constants.PREDICTION_LOG_KEY],
          ref_extracts[constants.PREDICTION_LOG_KEY],
      )
  
imports: import os
  import apache_beam
  from apache_beam.testing import util
  import numpy
  import tensorflow
  from tensorflow_model_analysis import constants
  from tensorflow_model_analysis.api import model_eval_lib
  from tensorflow_model_analysis.eval_saved_model import testutil
  from tensorflow_model_analysis.eval_saved_model.example_trainers import fixed_prediction_estimator_extra_fields
  from tensorflow_model_analysis.extractors import features_extractor
  from tensorflow_model_analysis.extractors import inference_base
  from tensorflow_model_analysis.extractors import tfx_bsl_predictions_extractor
  from tensorflow_model_analysis.proto import config_pb2
  from tfx_bsl.tfxio import tensor_adapter
  from tfx_bsl.tfxio import test_util
  from google.protobuf import text_format
  from tensorflow.core.protobuf import saved_model_pb2
  from tensorflow_metadata.proto.v0 import schema_pb2
  from tensorflow_serving.apis import logging_pb2
  from tensorflow_serving.apis import prediction_log_pb2
label_desc: Use image node_id 12 for calling the TfxBslPredictionsExtractorTest obj's underlying member method code with example usage: obj.testInsertTwoPredictionLogsIntoExtracts() without return types