code: def _test(self, input_data, src_format, dst_format, expected):
      for dtype in {np.int32, np.int64}:
          x = np.array(input_data, dtype=dtype)
          with self.session() as session:
              with self.test_scope():
                  placeholder = array_ops.placeholder(
                      dtypes.as_dtype(x.dtype), x.shape
                  )
                  param = {placeholder: x}
                  output = nn_ops.data_format_dim_map(
                      placeholder,
                      src_format=src_format,
                      dst_format=dst_format,
                  )
              result = session.run(output, param)
          self.assertAllEqual(result, expected)
  
imports: import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the XlaDataFormatDimMapTest obj's underlying member method code with example usage: obj._test(input_data, src_format, dst_format, expected) without return types