code: def setUp(self):
      super().setUp()
      self.correct_trajectory = jnp.array(
          [
              [dm_env.StepType.MID],
              [dm_env.StepType.MID],
              [dm_env.StepType.MID],
              [dm_env.StepType.LAST],
              [dm_env.StepType.FIRST],
              [dm_env.StepType.MID],
          ]
      )
      self.broken_trajectory = jnp.array(
          [
              [dm_env.StepType.MID],
              [dm_env.StepType.MID],
              [dm_env.StepType.MID],
              [dm_env.StepType.MID],
              [dm_env.StepType.FIRST],
              [dm_env.StepType.MID],
          ]
      )
  
file_path: rlax/rlax/_src/interruptions_test.py
imports: from absl.testing import absltest
  import dm_env
  import jax
  import jax.numpy
  import numpy
  from rlax._src import interruptions
label_desc: Use image node_id 1 for calling the InterruptionsTest obj's underlying member method code with example usage: obj.setUp() without return types