code: def test_conversion_with_policy(self, episode):
      rlds_data, expected_trajectories = self._valid_episodes[episode]
      generated_trajectories = (
          rlds_to_reverb.convert_rlds_to_trajectories(
              rlds_data, get_policy_info_test_fn
          )
      )  # type: tf.data.Dataset
      for generated_trajectory, expected_trajectory in zip(
          list(generated_trajectories.as_numpy_iterator()),
          expected_trajectories,
      ):
          self.assertEqual(
              generated_trajectory.step_type,
              expected_trajectory.step_type,
          )
          self.assertEqual(
              generated_trajectory.next_step_type,
              expected_trajectory.next_step_type,
          )
          self.assertAllEqual(
              generated_trajectory.observation,
              expected_trajectory.observation,
          )
          self.assertAllEqual(
              generated_trajectory.action, expected_trajectory.action
          )
          self.assertAllEqual(
              generated_trajectory.discount,
              expected_trajectory.discount,
          )
          self.assertAllEqual(
              generated_trajectory.reward, expected_trajectory.reward
          )
          self.assertAllEqual(
              generated_trajectory.policy_info,
              {rlds_types.ACTION: expected_trajectory.action},
          )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from typing import Dict, List, Tuple
  from absl.testing import parameterized
  import reverb
  from rlds import rlds_types
  import tensorflow
  from tf_agents.replay_buffers import reverb_replay_buffer
  from tf_agents.replay_buffers import reverb_utils
  from tf_agents.replay_buffers import rlds_to_reverb
  from tf_agents.specs import tensor_spec
  from tf_agents.trajectories import trajectory
  from tf_agents.utils import test_utils
label_desc: Use image node_id 5 for calling the RldsToTrajectoriesTest obj's underlying member method code with example usage: obj.test_conversion_with_policy(episode) without return types