code: def setUp(self):
      super().setUp()
      self._output_data_dir = os.path.join(
          os.environ.get(
              "TEST_UNDECLARED_OUTPUTS_DIR", self.get_temp_dir()
          ),
          self._testMethodName,
      )
  
file_path: tfx/tfx/components/example_gen/write_split_test.py
imports: import os
  import apache_beam
  from apache_beam.metrics.metric import MetricsFilter
  from apache_beam.runners.direct import direct_runner
  import pyarrow
  import tensorflow
  from tfx.components.example_gen import write_split
  from tfx.dsl.io import fileio
  from tfx.proto import example_gen_pb2
  from tfx.types import standard_component_specs
label_desc: Use image node_id 1 for calling the WriteSplitTest obj's underlying member method code with example usage: obj.setUp() without return types