code: def testGetColumn(self):
      self.assertTrue(
          arrow_util.get_column(_INPUT_RECORD_BATCH, "f1").equals(
              pa.array([[1], [2, 3]])
          )
      )
      self.assertIsNone(
          arrow_util.get_column(
              _INPUT_RECORD_BATCH, "xyz", missing_ok=True
          )
      )
      with self.assertRaises(KeyError):
          arrow_util.get_column(_INPUT_RECORD_BATCH, "xyz")
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import itertools
  from typing import Dict, Iterable, NamedTuple
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  import pyarrow
  import six
  from tensorflow_data_validation import types
  from tensorflow_data_validation.arrow import arrow_util
  from tensorflow_data_validation.utils.example_weight_map import ExampleWeightMap
  from tfx_bsl.arrow import array_util
  from tfx_bsl.types import tfx_namedtuple
label_desc: Use image node_id 10 for calling the ArrowUtilTest obj's underlying member method code with example usage: obj.testGetColumn() without return types