code: def test_shuffle_data_yields_all_samples(self, queue_size, n_samples):
      samples = iter(range(n_samples))
      shuffled_stream = data.shuffle(samples, queue_size)
      self.assertLen(list(shuffled_stream), n_samples)
  
imports: import itertools
  import os
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  from trax import data
label_desc: Use image node_id 3 for calling the InputsTest obj's underlying member method code with example usage: obj.test_shuffle_data_yields_all_samples(queue_size, n_samples) without return types