code: def test_select_masks():
      npolygons_range = (1, 10)
      nvertices_range = (3, 40)
      for batch_size in [1, 3]:
          for vertex_ndim in [2, 3, 6]:
              for vertex_dtype in [
                  np.float,
                  random.choice(
                      [np.int8, np.int16, np.int32, np.int64]
                  ),
              ]:
                  reindex_masks = random.choice([False, True])
                  yield (
                      check_select_masks,
                      batch_size,
                      npolygons_range,
                      nvertices_range,
                      vertex_ndim,
                      vertex_dtype,
                      reindex_masks,
                  )
  
file_path: DALI/dali/test/python/operator_2/test_segmentation_select_masks.py
imports: from nose_utils import assert_raises
  import numpy
  import nvidia.dali
  import nvidia.dali.fn
  import random
  from segmentation_test_utils import make_batch_select_masks
label_desc: Use image node_id 2 for calling a global function with example usage: test_select_masks() without return types