code: def setUp(self):
      super().setUp()
      self._home = self.tmp_dir
      self.enter_context(
          test_case_utils.override_env_var("HOME", self._home)
      )
      self.enter_context(
          test_case_utils.override_env_var(
              "AIRFLOW_HOME",
              os.path.join(os.environ["HOME"], "airflow"),
          )
      )
  
      # Flags for handler.
      self.engine = "airflow"
      self.pipeline_path = os.path.join(
          _testdata_dir, "test_pipeline_airflow_1.py"
      )
      self.pipeline_root = os.path.join(self._home, "tfx", "pipelines")
      self.pipeline_name = "chicago_taxi_simple"
      self.run_id = "manual__2019-07-19T19:56:02+00:00"
      self.runtime_parameter = {"a": "1", "b": "2"}
      self.runtime_parameter_json = json.dumps(self.runtime_parameter)
  
      self.enter_context(
          test_case_utils.change_working_dir(self.tmp_dir)
      )
  
      # Pipeline args for mocking subprocess
      self.pipeline_args = {labels.PIPELINE_NAME: self.pipeline_name}
      self._mock_get_airflow_version = self.enter_context(
          mock.patch.object(
              airflow_handler.AirflowHandler,
              "_get_airflow_version",
              return_value="2.0.1",
          )
      )
  
imports: import json
  import os
  import subprocess
  import sys
  from unittest import mock
  import click
  import tensorflow
  from tfx.dsl.components.base import base_driver
  from tfx.dsl.io import fileio
  from tfx.tools.cli import labels
  from tfx.tools.cli.handler import airflow_handler
  from tfx.utils import io_utils
  from tfx.utils import test_case_utils
label_desc: Use image node_id 1 for calling the AirflowHandlerTest obj's underlying member method code with example usage: obj.setUp() without return types