code: def setUp(self):
      super().setUp()
      old = jnp.zeros((3,), dtype=jnp.float32)
      new = jnp.ones((3,), dtype=jnp.float32)
      self._old_struct = ((old, old), old)
      self._new_struct = ((new, new), new)
  
file_path: rlax/rlax/_src/nested_updates_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import chex
  import jax
  import jax.numpy
  import numpy
  from rlax._src import nested_updates
label_desc: Use image node_id 1 for calling the NestedUpdatesTest obj's underlying member method code with example usage: obj.setUp() without return types