code: def setUp(self):
      super().setUp()
      self.potential_discount = 0.99
  
      self.features = np.array(
          [
              [[1.0, 2.0], [2.0, 4.0], [3.0, 6.0]],
              [[1.0, 2.0], [0.0, 0.0], [-1.0, -2.0]],
          ]
      )
  
      self.exp_feature = np.array(
          [[[2.0, 4.0], [3.0, 6.0]], [[0.0, 0.0], [-1.0, -2.0]]]
      )
      self.exp_abs_change = np.array(
          [[[1.0, 2.0], [1.0, 2.0]], [[1.0, 2.0], [1.0, 2.0]]]
      )
      self.exp_increase = np.array(
          [[[1.0, 2.0], [1.0, 2.0]], [[-1.0, -2.0], [-1.0, -2.0]]]
      )
      self.exp_decrease = np.array(
          [[[-1.0, -2.0], [-1.0, -2.0]], [[1.0, 2.0], [1.0, 2.0]]]
      )
  
      g = self.potential_discount
      self.exp_potential = np.array(
          [
              [
                  [g * 2.0 - 1.0, g * 4.0 - 2.0],
                  [g * 3.0 - 2.0, g * 6.0 - 4.0],
              ],
              [[-1.0, -2.0], [-g, -2.0 * g]],
          ]
      )
  
file_path: rlax/rlax/_src/general_value_functions_test.py
imports: import functools
  from absl.testing import absltest
  from absl.testing import parameterized
  import chex
  import jax
  import numpy
  from rlax._src import general_value_functions
label_desc: Use image node_id 1 for calling the FeatureControlTest obj's underlying member method code with example usage: obj.setUp() without return types