code: def testStoreTransitionWithPrioritizedSampling(self):
      agent = full_rainbow_agent.JaxFullRainbowAgent(
          num_actions=4, replay_scheme="prioritized"
      )
      dummy_frame = onp.zeros((84, 84))
      # Adding transitions with default, 10., default priorities.
      agent._store_transition(dummy_frame, 0, 0, False)
      agent._store_transition(dummy_frame, 0, 0, False, priority=10.0)
      agent._store_transition(dummy_frame, 0, 0, False)
      returned_priorities = agent._replay.get_priority(
          onp.arange(
              self.stack_size - 1, self.stack_size + 2, dtype=onp.int32
          )
      )
      expected_priorities = [1.0, 10.0, 10.0]
      onp.array_equal(returned_priorities, expected_priorities)
  
file_path: dopamine/tests/dopamine/jax/agents/full_rainbow/full_rainbow_agent_test.py
imports: from absl.testing import absltest
  from dopamine.discrete_domains import atari_lib
  from dopamine.jax.agents.dqn import dqn_agent
  from dopamine.jax.agents.full_rainbow import full_rainbow_agent
  from dopamine.utils import test_utils
  from flax import linen
  import gin
  import jax.numpy
  import numpy
label_desc: Use image node_id 9 for calling the FullRainbowAgentTest obj's underlying member method code with example usage: obj.testStoreTransitionWithPrioritizedSampling() without return types