code: def test_columns__with_columns_set(self):
      kwargs = [
          {"images": [0]},
          {"heatmaps": [0]},
          {"segmentation_maps": [0]},
          {"keypoints": [0]},
          {"bounding_boxes": [0]},
          {"polygons": [0]},
          {"line_strings": [0]},
      ]
      for kwargs_i in kwargs:
          batch = _BatchInAugmentation(**kwargs_i)
          columns = batch.columns
          assert len(columns) == 1
          assert columns[0].name == list(kwargs_i.keys())[0]
  
imports: from __future__ import print_function, division, absolute_import
  import warnings
  import sys
  import numpy
  import imgaug
  import imgaug.augmenters
  from imgaug.testutils import reseed
  from imgaug.augmentables.batches import _BatchInAugmentation
label_desc: Use image node_id 9 for calling the Test_BatchInAugmentation obj's underlying member method code with example usage: obj.test_columns__with_columns_set() without return types