code: def test_simple(self):
      m = batch_apply.BatchApply(AddOne())
      x = tf.zeros([2, 3, 4])
      y = m(x)
      self.assertAllEqual(y, tf.ones([2, 3, 4]))
  
imports: from absl.testing import parameterized
  import numpy
  from sonnet.src import base
  from sonnet.src import batch_apply
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 1 for calling the BatchApplyTest obj's underlying member method code with example usage: obj.test_simple() without return types