code: def test_init(self, BatchNormLayer, init_unique):
      input_shape = (2, 3, 4)
      # default: normalize over all but second axis
      beta = BatchNormLayer(input_shape, beta=init_unique).beta
      assert np.allclose(beta.get_value(), init_unique((3,)))
      # normalize over first axis only
      beta = BatchNormLayer(input_shape, beta=init_unique, axes=0).beta
      assert np.allclose(beta.get_value(), init_unique((3, 4)))
      # normalize over second and third axis
      try:
          beta = BatchNormLayer(
              input_shape, beta=init_unique, axes=(1, 2)
          ).beta
          assert np.allclose(beta.get_value(), init_unique((2,)))
      except ValueError as exc:
          assert "BatchNormDNNLayer only supports" in exc.args[0]
  
imports: from mock import Mock
  import numpy
  import pytest
  import theano
label_desc: Use image node_id 3 for calling the TestBatchNormLayer obj's underlying member method code with example usage: obj.test_init(BatchNormLayer, init_unique) without return types