code: def testQValueComputation(self):
      agent = self._create_test_agent()
      quantiles = onp.ones(agent.num_quantile_samples)
      q_value = onp.sum(quantiles)
      quantiles = quantiles.reshape([agent.num_quantile_samples, 1])
      expected_q_values = onp.ones([agent.num_actions]) * q_value
      for params in [agent.online_params, agent.target_network_params]:
          agent._rng, rng_input = jax.random.split(agent._rng)
          q_values = jnp.mean(
              agent.network_def.apply(
                  params,
                  self.ones_state,
                  num_quantiles=agent.num_quantile_samples,
                  rng=rng_input,
              ).quantile_values,
              axis=0,
          )
          onp.array_equal(q_values, expected_q_values)
          self.assertEqual(jnp.argmax(q_values, axis=0), 0)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import absltest
  from dopamine.discrete_domains import atari_lib
  from dopamine.jax.agents.dqn import dqn_agent
  from dopamine.jax.agents.implicit_quantile import implicit_quantile_agent
  from flax import linen
  import gin
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 5 for calling the ImplicitQuantileAgentTest obj's underlying member method code with example usage: obj.testQValueComputation() without return types