Example #1
0
def test_not_find_armnn(tmpdir):
    with pytest.raises(RuntimeError) as err:
        find_armnn(lib_name='libarmnn*.so',
                   armnn_libs_env="RUBBISH_LIB",
                   default_lib_search=("/lib", ))

    assert 'ArmNN library libarmnn*.so was not found in (\'/lib\',)' in str(
        err.value)
Example #2
0
def test_find_armnn_default_path(tmpdir):
    lib_names, lib_paths = find_armnn(
        lib_name='libarmnn*.so',
        armnn_libs_env="RUBBISH_LIB",
        default_lib_search=(os.environ["TEST_ARMNN_LIB"], ))
    armnn_includes = find_includes('TEST_ARMNN_INCLUDE')
    assert [
        ':libarmnn.so', ':libarmnnSomeThing1.so', ':libarmnnSomeThing2.so'
    ] == sorted(lib_names)
    assert [os.path.join(tmpdir, 'lib')] == lib_paths
    assert [os.path.join(tmpdir, 'include')] == armnn_includes
Example #3
0
def test_find_armnn(tmpdir):
    lib_names, lib_paths = find_armnn(lib_name='libarmnn*.so',
                                      armnn_libs_env="TEST_ARMNN_LIB",
                                      default_lib_search=("/lib", ))
    armnn_includes = find_includes(armnn_include_env="TEST_ARMNN_INCLUDE")

    assert [
        ':libarmnn.so', ':libarmnnSomeThing1.so', ':libarmnnSomeThing2.so'
    ] == sorted(lib_names)
    assert [os.path.join(tmpdir, 'lib')] == lib_paths
    assert [os.path.join(tmpdir, 'include')] == armnn_includes