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']