docstr: Check periodic update enabled.
docstr_tok: ["Check","periodic","update","enabled","."]
code: def test_conditional_update_is_time(self):
      """Check periodic update enabled."""
      conditional_update = self.variant(
          nested_updates.conditional_update
      )
  
      is_time = jnp.array(True)
      output = conditional_update(
          self._new_struct, self._old_struct, is_time
      )
      for o, exp in zip(
          jax.tree_leaves(output), jax.tree_leaves(self._new_struct)
      ):
          np.testing.assert_allclose(o, exp)
  
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 2 for calling the NestedUpdatesTest obj's underlying member method code with example usage: obj.test_conditional_update_is_time() without return types