code: def testCustomBeamMakePipelineFn(self):
      mock_fn = mock.MagicMock()
      executor_context = base_beam_executor.BaseBeamExecutor.Context(
          make_beam_pipeline_fn=mock_fn
      )
      executor = _TestExecutor(executor_context)
      executor._make_beam_pipeline()
      mock_fn.assert_called_once_with()
  
file_path: tfx/tfx/dsl/components/base/base_beam_executor_test.py
imports: import sys
  from typing import Any, Dict, List
  from unittest import mock
  from apache_beam.options.pipeline_options import DirectOptions
  from apache_beam.options.pipeline_options import GoogleCloudOptions
  from apache_beam.options.pipeline_options import StandardOptions
  import tensorflow
  from tfx import types
  from tfx import version
  from tfx.components.statistics_gen.executor import Executor
  from tfx.dsl.components.base import base_beam_executor
label_desc: Use image node_id 2 for calling the BaseBeamExecutorTest obj's underlying member method code with example usage: obj.testCustomBeamMakePipelineFn() without return types