code: def test_new_weights_and_state(self):
      layer = tl.BatchNorm()
      x = np.ones((3, 2, 7)).astype(np.float32)
      _, _ = layer.init(shapes.signature(x))
  
      running_mean, running_var, n_batches = layer.state
      np.testing.assert_allclose(running_mean, 0.0)
      np.testing.assert_allclose(running_var, 1.0)
      self.assertEqual(n_batches, 0)
  
file_path: trax/trax/layers/normalization_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  from trax import fastmath
  from trax import shapes
  import trax.layers
label_desc: Use image node_id 4 for calling the BatchNormTest obj's underlying member method code with example usage: obj.test_new_weights_and_state() without return types