code: def test_zero_decay(self):
      ema = moving_averages.ExponentialMovingAverage(0.0)
      random_input = jax.random.uniform(
          jax.random.PRNGKey(428), shape=(2, 3, 4)
      )
  
      # The ema should be equal to the input with decay=0.
      np.testing.assert_allclose(random_input[0], ema(random_input[0]))
      np.testing.assert_allclose(random_input[1], ema(random_input[1]))
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import basic
  from haiku._src import moving_averages
  from haiku._src import multi_transform
  from haiku._src import test_utils
  from haiku._src import transform
  import jax
  import jax.numpy
  import jax.random
  import numpy
  import tree
label_desc: Use image node_id 1 for calling the MovingAveragesTest obj's underlying member method code with example usage: obj.test_zero_decay() without return types