code: def test_unnormalize_linear(self):
      num_outputs = 3
      state = get_fake_pop_art_state(num_outputs)
  
      # Verify that it denormalizes as planned.
      inputs = np.ones([num_outputs, num_outputs])
      indices = np.arange(num_outputs)[::-1]
      out = pop_art.unnormalize_linear(state, inputs, indices)
  
      expected_normalized = np.asarray([1, 1, 1])
      expected_unnormalized = np.asarray([6, 4, 2])
      np.testing.assert_allclose(out.normalized, expected_normalized)
      np.testing.assert_allclose(
          out.unnormalized, expected_unnormalized
      )
  
imports: import functools
  from absl.testing import absltest
  from absl.testing import parameterized
  import chex
  import haiku
  from haiku import data_structures
  from haiku import initializers
  import jax
  import jax.nn
  import jax.numpy
  import numpy
  from rlax._src import pop_art
label_desc: Use image node_id 3 for calling the PopArtTest obj's underlying member method code with example usage: obj.test_unnormalize_linear() without return types