code: def _mock_build_graph_fn(self, store, input_graph):
      del store  # Unused.
      graph_key = text_format.MessageToString(
          input_graph, as_one_line=True
      )
      return self._graph_fn_result[graph_key]
  
imports: from typing import Set
  from unittest import mock
  import tensorflow
  from tfx.orchestration.portable.input_resolution import exceptions
  from tfx.orchestration.portable.input_resolution import input_graph_resolver
  from tfx.orchestration.portable.input_resolution import node_inputs_resolver
  from tfx.orchestration.portable.input_resolution import partition_utils
  from tfx.orchestration.portable.input_resolution import channel_resolver
  from tfx.proto.orchestration import pipeline_pb2
  import tfx.types
  from tfx.types import channel_utils
  from tfx.utils import test_case_utils
  from google.protobuf import text_format
label_desc: Use image node_id 5 for calling the NodeInputsResolverTest obj's underlying member method code with example usage: obj._mock_build_graph_fn(store, input_graph) and returns: self