class_docstr: Tests input-output pairs for the SpaceToBatch and BatchToSpace ops.
code: def testSmallInput2x2(self):
      x_np = [[[[1], [2]], [[3], [4]]]]
      block_size = 2
      x_out = [[[[1]]], [[[2]]], [[[3]]], [[[4]]]]
      self._testOne(x_np, 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 3 for calling the SpaceToBatchTest obj's underlying member method code with example usage: obj.testSmallInput2x2() without return types