code: def test_sorted_bucket_sampler(
      seq_lengths, mult, batch_size, shuffle
  ):
      sampler = s.SortedBucketSampler(
          sort_keys=seq_lengths,
          batch_size=batch_size,
          mult=mult,
          shuffle=shuffle,
      )
      total_sampled_ids = []
      for batch_sample_ids in sampler:
          total_sampled_ids.extend(batch_sample_ids)
      assert len(set(total_sampled_ids)) == len(total_sampled_ids) == N
  
file_path: gluon-nlp/tests/test_data_sampler.py
imports: import pytest
  import numpy
  from mxnet.gluon import data
  from gluonnlp.data import sampler
label_desc: Use image node_id 6 for calling a global function with example usage: test_sorted_bucket_sampler(seq_lengths, mult, batch_size, shuffle) without return types