code: def test_conditional_split_merge():
      rng = np.random.default_rng()
      for dev in ["cpu", "gpu"]:
          for pred_gen in [
              lambda x: np.array(x < 3),
              lambda x: np.array(x % 2 == 0),
              lambda x: np.array(x % 3 == 0),
              lambda _: np.array(False),
              lambda _: rng.choice([np.array(True), np.array(False)]),
          ]:
              yield check_conditional_split_merge, dev, pred_gen
  
file_path: DALI/dali/test/python/operator_2/test_split_merge.py
imports: from nvidia.dali.pipeline import pipeline_def, Pipeline
  import nvidia.dali.fn
  import nvidia.dali.types
  import numpy
  from test_utils import check_batch, RandomlyShapedDataIterator
  from nose_utils import assert_raises
  from nose2.tools import params
label_desc: Use image node_id 6 for calling a global function with example usage: test_conditional_split_merge() without return types