code: def test_uniform(self, i, args):
      device, policy_name, batch_size, use_shape = args
      num_iterations = 3
  
      assert device in ("gpu", "cpu")
  
      @pipeline_def(
          batch_size=batch_size,
          device_id=0,
          num_threads=4,
          seed=205,
          enable_conditionals=True,
      )
      def pipeline():
          rng = random.Random(42 + i)
          video = fn.external_source(
              source=lambda: list(
                  rng.choices(self.vid_files, k=batch_size)
              ),
              batch=True,
              layout="FHWC",
          )
          extra = (
              {} if not use_shape else {"shape": fn.shapes(video)[1:]}
          )
          if device == "gpu":
              video = video.gpu()
          video = auto_augment.auto_augment(video, policy_name, **extra)
          return video
  
      # run the pipeline twice to make sure instantiation preserves determinism
      p1 = pipeline()
      p1.build()
      p2 = pipeline()
      p2.build()
  
      for _ in range(num_iterations):
          (out1,) = p1.run()
          (out2,) = p2.run()
          check_batch(out1, out2)
  
imports: import itertools
  import os
  import unittest
  import random
  import numpy
  from scipy.stats import chisquare
  from nose2.tools import params
  from nvidia.dali import fn, types
  from nvidia.dali import pipeline_def
  from nvidia.dali.auto_aug import auto_augment, augmentations
  from nvidia.dali.auto_aug.core import augmentation, Policy
  from test_utils import get_dali_extra_path, check_batch
  from nose_utils import assert_raises, assert_warns
label_desc: Use image node_id 2 for calling the VideoTest obj's underlying member method code with example usage: obj.test_uniform(i, args) and returns: video