1# RUN: %PYTHON %s 2>&1 2 3import os 4 5from mlir._mlir_libs import get_include_dirs, get_lib_dirs 6 7 8header_file = os.path.join(get_include_dirs()[0], "mlir-c", "IR.h") 9assert os.path.isfile(header_file), f"Header does not exist: {header_file}" 10 11# Since actual library names are platform specific, just scan the directory 12# for a filename that contains the library name. 13expected_lib_name = "MLIRPythonCAPI" 14all_libs = os.listdir(get_lib_dirs()[0]) 15found_lib = False 16for file_name in all_libs: 17 if expected_lib_name in file_name: 18 found_lib = True 19assert found_lib, f"Did not find '{expected_lib_name}' lib in {all_libs}" 20