docstr: Test that the dataset reader iterates over batches of rows correctly.
docstr_tok: ["Test","that","the","dataset","reader","iterates","over","batches","of","rows","correctly","."]
code: def test_dataset_reader_itr_batches(self):
      """Test that the dataset reader iterates over batches of rows correctly."""
      input_config = {"format": "json", "paths": self.dset_path}
      dataset, _ = get_dataset_and_shards(
          AlgorithmConfig().offline_data(
              input_="dataset", input_config=input_config
          )
      )
  
      ioctx = IOContext(
          config=(
              AlgorithmConfig()
              .training(train_batch_size=1200)
              .offline_data(actions_in_input_normalized=True)
          ),
          worker_index=0,
      )
      reader = DatasetReader(dataset, ioctx)
      assert len(reader.next()) >= 1200
  
imports: import tempfile
  import os
  from pathlib import Path
  import unittest
  import pytest
  import ray
  from ray.rllib.algorithms.algorithm_config import AlgorithmConfig
  from ray.rllib.offline import IOContext
  from ray.rllib.offline.dataset_reader import DatasetReader, get_dataset_and_shards, _unzip_if_needed
label_desc: Use image node_id 3 for calling the TestDatasetReader obj's underlying member method code with example usage: obj.test_dataset_reader_itr_batches() without return types