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
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
"-o {} ".format(os.path.join(out_dir, f"{name}_wrap.cpp")) +
f"-outdir {out_dir} " + f"{extr_includes} " + f"-I{in_dir} " +
os.path.join(in_dir, f"{name}.i"))
if code != 0:
raise RuntimeError(f"Failed to generate {name} ext.")
if __name__ == "__main__":
__swig_exec = get_swig_exec()
# This check is redundant in case CMake is used, it's here for standalone use
if not check_swig_version('4.'):
raise RuntimeError(
"Wrong swig version was found. Expected SWIG version is 4.x.x")
armnn_includes = find_includes()
parser = argparse.ArgumentParser("Script to generate SWIG wrappers.")
parser.add_argument("-v",
"--verbose",
help="Verbose output.",
action="store_true")
args = parser.parse_args()
__verbose = args.verbose
wrap_names = [
'armnn_version', 'armnn', 'armnn_caffeparser', 'armnn_onnxparser',
'armnn_tfparser', 'armnn_tfliteparser'
]
def test_rubbish_armnn_include(tmpdir, env):
includes = find_includes(armnn_include_env=env)
assert includes == ['/usr/local/include', '/usr/include']
