class_docstr: ArrowDatasetTest
class_docstr_tok: ["ArrowDatasetTest"]
docstr: test_arrow_dataset
code: def test_arrow_dataset(self):
      """test_arrow_dataset"""
      import tensorflow_io.arrow as arrow_io
  
      truth_data = TruthData(
          self.scalar_data + self.list_data,
          self.scalar_dtypes + self.list_dtypes,
          self.scalar_shapes + self.list_shapes,
      )
  
      batch = self.make_record_batch(truth_data)
  
      # test all columns selected
      dataset = arrow_io.ArrowDataset.from_record_batches(
          batch, truth_data.output_types, truth_data.output_shapes
      )
      self.run_test_case(dataset, truth_data)
  
      # test column selection
      columns = (1, 3, len(truth_data.output_types) - 1)
      dataset = arrow_io.ArrowDataset.from_record_batches(
          batch,
          tuple(truth_data.output_types[c] for c in columns),
          tuple(truth_data.output_shapes[c] for c in columns),
          columns=columns,
      )
      self.run_test_case(dataset, truth_data)
  
      # test construction from pd.DataFrame
      df = batch.to_pandas()
      dataset = arrow_io.ArrowDataset.from_pandas(
          df, preserve_index=False
      )
      self.run_test_case(dataset, truth_data)
  
imports: from collections import namedtuple
  import io
  import os
  import socket
  import tempfile
  import threading
  import pytest
  import pyarrow
  import numpy.testing
  import tensorflow
  import tensorflow_io
label_desc: Use image node_id 2 for calling the ArrowDatasetTest obj's underlying member method code with example usage: obj.test_arrow_dataset() without return types