code: def test_episodic_dataset(self):
      pass
      num_batches = 0
      for batch in self.dataset:
          num_batches += 1
          self.assertLessEqual(len(batch["reward"]), self.max_steps)
          self.assertIsInstance(batch, dict)
      self.assertEqual(num_batches, self.num_episodes)
  
file_path: ReAgent/reagent/gym/tests/test_gym_datasets.py
imports: import logging
  import unittest
  from reagent.gym.agents.agent import Agent
  from reagent.gym.datasets.episodic_dataset import EpisodicDataset
  from reagent.gym.envs import Gym
  from reagent.gym.policies.policy import Policy
  from reagent.gym.policies.samplers.discrete_sampler import SoftmaxActionSampler
  from reagent.gym.utils import build_normalizer
  from reagent.net_builder.discrete_dqn.fully_connected import FullyConnected
label_desc: Use image node_id 2 for calling the TestEpisodicDataset obj's underlying member method code with example usage: obj.test_episodic_dataset() without return types