code: def test_batchnormalization(self):
      K.set_image_dim_ordering("th")
      input_data = np.random.random_sample([2, 5, 32, 32])
      blayer = BLayer.BatchNormalization(
          axis=1, input_shape=(5, 32, 32)
      )
      klayer = KLayer.BatchNormalization(
          axis=1, input_shape=(5, 32, 32)
      )
      self.compare_newapi(
          klayer,
          blayer,
          input_data,
          WeightsConverter.convert_batchnormalization,
      )
      K.set_image_dim_ordering("tf")
      input_data2 = np.random.random_sample([2, 32, 32, 4])
      blayer = BLayer.BatchNormalization(
          axis=-1, dim_ordering="tf", input_shape=(32, 32, 4)
      )
      klayer = KLayer.BatchNormalization(
          axis=-1, input_shape=(32, 32, 4)
      )
      self.compare_newapi(
          klayer,
          blayer,
          input_data2,
          WeightsConverter.convert_batchnormalization,
      )
  
imports: import pytest
  from test.bigdl.test_utils import BigDLTestCase
  import bigdl.dllib.nn.keras.layers.layer
  import keras.layers
  import keras.backend
  from bigdl.dllib.keras.converter import WeightsConverter
  from bigdl.dllib.feature.dataset.dataset import *
  from bigdl.dllib.nn.keras.layers.topology import Model
  from bigdl.dllib.nn.keras.layers.topology import Sequential
  from keras.engine import merge, Model
  from keras.models import Sequential
label_desc: Use image node_id 2 for calling the TestKerasAPI obj's underlying member method code with example usage: obj.test_batchnormalization() without return types