code: def test_non_dist(self, mock):
      sampler = BatchAugSampler(
          self.dataset, num_repeats=3, shuffle=False
      )
      self.assertEqual(sampler.world_size, 1)
      self.assertEqual(sampler.rank, 0)
      self.assertEqual(sampler.total_size, self.data_length * 3)
      self.assertEqual(sampler.num_samples, self.data_length * 3)
      indices = [x for x in range(self.data_length) for _ in range(3)]
      self.assertEqual(list(sampler), indices)
  
file_path: mmocr/tests/test_datasets/test_samplers/test_batch_aug.py
imports: from unittest import TestCase
  from unittest.mock import patch
  import torch
  from mmengine.logging import MMLogger
  from mmocr.datasets import BatchAugSampler
label_desc: Use image node_id 2 for calling the TestBatchAugSampler obj's underlying member method code with example usage: obj.test_non_dist(mock) without return types