code: def test_normalize_collect(self):
      x = np.random.uniform(size=(2, 3, 4, 5))
      normalize = normalization.Normalize(mode="collect")
      normalize.init(shapes.signature(x))
      old_state = normalize.state
      y = normalize(x)
      with self.assertRaises(AssertionError):
          np.testing.assert_equal(normalize.state, old_state)
      with self.assertRaises(AssertionError):
          np.testing.assert_almost_equal(x, y)
  
file_path: trax/trax/rl/normalization_test.py
imports: from absl.testing import absltest
  import numpy
  from trax import shapes
  from trax.rl import normalization
label_desc: Use image node_id 3 for calling the NormalizationTest obj's underlying member method code with example usage: obj.test_normalize_collect() without return types