class_docstr: Tests input-output pairs for the SpaceToBatch and BatchToSpace ops.
code: def testSmallInput2x2Pad1x0(self):
      x_np = [[[[1], [2]], [[3], [4]]]]
      paddings = np.array([[1, 0], [1, 0]], dtype=np.int32)
      block_size = 3
      x_out = [
          [[[0]]],
          [[[0]]],
          [[[0]]],
          [[[0]]],
          [[[1]]],
          [[[2]]],
          [[[0]]],
          [[[3]]],
          [[[4]]],
      ]
      self._testPad(x_np, paddings, block_size, x_out)
  
imports: import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gen_array_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 4 for calling the SpaceToBatchTest obj's underlying member method code with example usage: obj.testSmallInput2x2Pad1x0() without return types