code: def test_dist(self, mock):
      sampler = BatchAugSampler(
          self.dataset, num_repeats=3, shuffle=False
      )
      self.assertEqual(sampler.world_size, 3)
      self.assertEqual(sampler.rank, 2)
      self.assertEqual(sampler.num_samples, self.data_length)
      self.assertEqual(sampler.total_size, self.data_length * 3)
  
      logger = MMLogger.get_current_instance()
      with patch.object(logger, "warning") as mock_log:
          sampler = BatchAugSampler(self.dataset, shuffle=False)
          mock_log.assert_not_called()
  
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 3 for calling the TestBatchAugSampler obj's underlying member method code with example usage: obj.test_dist(mock) without return types