def METHOD_NAME():
    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