code: def setUp(self):
      super().setUp()
      self.r_t = np.array(
          [[1.0, 0.0, -1.0, 0.0, 1.0], [0.5, 0.8, -0.7, 0.0, 2.1]]
      )
      self.discount_t = np.array(
          [[0.5, 0.9, 1.0, 0.5, 0.8], [0.9, 0.5, 0.3, 0.8, 0.7]]
      )
      self.v_t = np.array(
          [[3.0, 1.0, 5.0, -5.0, 3.0], [-1.7, 1.2, 2.3, 2.2, 2.7]]
      )
  
      # Different expected results for different values of n.
      self.expected = {}
      self.expected[3] = np.array(
          [
              [2.8, -3.15, 0.7, 1.7, 3.4],
              [1.2155, 0.714, 0.2576, 3.192, 3.99],
          ],
          dtype=np.float32,
      )
      self.expected[5] = np.array(
          [
              [1.315, 0.63, 0.7, 1.7, 3.4],
              [1.33592, 0.9288, 0.2576, 3.192, 3.99],
          ],
          dtype=np.float32,
      )
      self.expected[7] = np.array(
          [
              [1.315, 0.63, 0.7, 1.7, 3.4],
              [1.33592, 0.9288, 0.2576, 3.192, 3.99],
          ],
          dtype=np.float32,
      )
  
imports: import functools
  from absl.testing import absltest
  from absl.testing import parameterized
  import chex
  import jax
  import jax.numpy
  import numpy
  from rlax._src import multistep
label_desc: Use image node_id 1 for calling the NStepBootstrappedReturnsTest obj's underlying member method code with example usage: obj.setUp() without return types