Function test_api_fw_check1

dali/test/python/test_fw_iterators_detection.py:98–131

(iter_type, data_definition)

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


@attr("pytorch")