code: def _test_number_of_batches(self, data_iterator, expected_batches):
      num_batches = 0
      for _ in data_iterator:
          num_batches += 1
      self.assertEqual(num_batches, expected_batches)
  
imports: import unittest
  from classy_vision.tasks import build_task
  from test.generic.config_utils import get_test_task_config
label_desc: Use image node_id 1 for calling the TestDataloaderLimitWrapper obj's underlying member method code with example usage: obj._test_number_of_batches(data_iterator, expected_batches) without return types