code: def mock_graph_fn_result(
      self, input_graph, graph_fn, dependent_inputs=()
  ):
      graph_key = text_format.MessageToString(
          input_graph, as_one_line=True
      )
      self._graph_fn_result[graph_key] = (graph_fn, dependent_inputs)
  
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 3 for calling the NodeInputsResolverTest obj's underlying member method code with example usage: obj.mock_graph_fn_result(input_graph, graph_fn, dependent_inputs) without return types