code: def setUp(self):
      super().setUp()
      self.q_t = np.array(
          [
              [[1.2, 2.2], [-1.2, 0.2], [2.2, -1.2]],
              [[4.2, 2.2], [1.2, 1.2], [-1.2, -2.2]],
          ],
          dtype=np.float64,
      )
  
imports: import functools
  from absl.testing import absltest
  from absl.testing import parameterized
  import chex
  import jax
  import numpy
  from rlax._src import nonlinear_bellman
label_desc: Use image node_id 1 for calling the IdentityTest obj's underlying member method code with example usage: obj.setUp() without return types