code: def _write_records_to_csv(
      self, records, tmp_dir, filename, compression_type=""
  ):
      data_location = os.path.join(tmp_dir, filename)
      if compression_type == "gzip":
          with gzip.GzipFile(data_location, "wb") as writer:
              writer.write("\n".join(records).encode("utf-8"))
      else:
          with open(data_location, "w") as writer:
              writer.write("\n".join(records))
      return data_location
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import gzip
  import os
  import tempfile
  from absl.testing import absltest
  from absl.testing import parameterized
  from apache_beam.io.filesystem import CompressionTypes
  import pandas
  import tensorflow
  from tensorflow_data_validation.statistics import stats_options
  from tensorflow_data_validation.utils import stats_gen_lib
  from tensorflow_data_validation.utils import test_util
  from google.protobuf import text_format
  from tensorflow_metadata.proto.v0 import schema_pb2
  from tensorflow_metadata.proto.v0 import statistics_pb2
label_desc: Use image node_id 6 for calling the StatsGenTest obj's underlying member method code with example usage: obj._write_records_to_csv(records, tmp_dir, filename, compression_type) and returns: data_location