code: def setUp(self):
      super().setUp()
  
      # Create a pipeline IR containing deployment config for testing.
      deployment_config = pipeline_pb2.IntermediateDeploymentConfig()
      executor_spec = pipeline_pb2.ExecutorSpec.PythonClassExecutorSpec(
          class_path="trainer.TrainerExecutor"
      )
      deployment_config.executor_specs["Trainer"].Pack(executor_spec)
      deployment_config.executor_specs["Transform"].Pack(executor_spec)
      deployment_config.executor_specs["Evaluator"].Pack(executor_spec)
      deployment_config.executor_specs["Pusher"].Pack(executor_spec)
      pipeline = pipeline_pb2.Pipeline()
      pipeline.nodes.add().pipeline_node.node_info.id = "Trainer"
      pipeline.nodes.add().pipeline_node.node_info.id = "Transform"
      pipeline.nodes.add().pipeline_node.node_info.id = "Evaluator"
      pipeline.nodes.add().pipeline_node.node_info.id = "Pusher"
      pipeline.pipeline_info.id = "test-pipeline"
      pipeline.deployment_config.Pack(deployment_config)
  
      ts.TaskSchedulerRegistry.clear()
  
      self._deployment_config = deployment_config
      self._pipeline = pipeline
      self._type_url = deployment_config.executor_specs[
          "Trainer"
      ].type_url
  
imports: import contextlib
  import functools
  import os
  import threading
  import time
  from absl import logging
  from absl.testing.absltest import mock
  import tensorflow
  from tfx.orchestration import data_types_utils
  from tfx.orchestration import metadata
  from tfx.orchestration.experimental.core import async_pipeline_task_gen
  from tfx.orchestration.experimental.core import constants
  from tfx.orchestration.experimental.core import pipeline_state
  from tfx.orchestration.experimental.core import post_execution_utils
  from tfx.orchestration.experimental.core import service_jobs
  from tfx.orchestration.experimental.core import task
  from tfx.orchestration.experimental.core import task_manager
  from tfx.orchestration.experimental.core import task_queue
  from tfx.orchestration.experimental.core import task_scheduler
  from tfx.orchestration.experimental.core import test_utils
  from tfx.orchestration.experimental.core.testing import test_async_pipeline
  from tfx.orchestration import mlmd_connection_manager
  from tfx.proto.orchestration import execution_result_pb2
  from tfx.proto.orchestration import pipeline_pb2
  from tfx.utils import status
  from ml_metadata.proto import metadata_store_pb2
label_desc: Use image node_id 1 for calling the TaskManagerTest obj's underlying member method code with example usage: obj.setUp() without return types