Function test_api_fw_check2

dali/test/python/test_fw_iterators_detection.py:149–179

(iter_type, data_definition)

@nottest
def test_api_fw_check2(iter_type, data_definition):
    root, annotations = data_paths()

    pipe = DetectionPipeline(BATCH_SIZE, 0, root, annotations)
    pipe.schedule_run()
    pipe.share_outputs()
    pipe.release_outputs()
    pipe.schedule_run()
    pipe.outputs()
    with assert_raises(
        RuntimeError,
        glob=(
            "Mixing pipeline API type. Currently used: PipelineAPIType.SCHEDULED,"
            " but trying to use PipelineAPIType.ITERATOR"
        ),
    ):
        train_loader = iter_type(
            [pipe], data_definition, EPOCH_SIZE, auto_reset=False, dynamic_shape=True
        )
        train_loader.__next__()
    # disable check
    pipe.enable_api_check(False)
    try:
        train_loader = iter_type(
            [pipe], data_definition, EPOCH_SIZE, auto_reset=False, dynamic_shape=True
        )
        train_loader.__next__()
        assert True
    except RuntimeError:
        assert False
    yield check, iter_type


def check(iter_type):