code: def test_compare_with_baseline(self):
      def batch_by_size_fn_wrapper(
          indices,
          num_tokens_vec,
          max_tokens,
          max_sentences,
          bsz_mult,
      ):
          def num_tokens_fn(idx):
              return num_tokens_vec[idx]
  
          return batch_by_size_fn(
              indices,
              num_tokens_fn,
              max_tokens,
              max_sentences,
              bsz_mult,
          )
  
      self._run_compare_with_baseline_sweep(batch_by_size_fn_wrapper)
  
file_path: fairseq/tests/test_data_utils.py
imports: import unittest
  import numpy
  from fairseq.data.data_utils_fast import batch_by_size_fn, batch_by_size_vec
label_desc: Use image node_id 1 for calling the TestBatchBySizeFn obj's underlying member method code with example usage: obj.test_compare_with_baseline() and returns: batch_by_size_fn, num_tokens_vec