docstr: Tests that tx(inv_tx(x)) == inv_tx(tx(x)) == x.
docstr_tok: ["Tests","that","tx","(","inv_tx","(","x",")",")","==","inv_tx","(","tx","(","x",")",")","==","x","."]
code: def test_identity(self, tx, inv_tx):
      """Tests that tx(inv_tx(x)) == inv_tx(tx(x)) == x."""
      np.testing.assert_allclose(
          inv_tx(tx(self.q_t)), self.q_t, rtol=1e-3
      )
      np.testing.assert_allclose(
          tx(inv_tx(self.q_t)), self.q_t, rtol=1e-3
      )
  
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 2 for calling the IdentityTest obj's underlying member method code with example usage: obj.test_identity(tx, inv_tx) without return types