15b2e7f50SStella Laurenzo# RUN: %PYTHON %s 2>&1 25b2e7f50SStella Laurenzo 35b2e7f50SStella Laurenzoimport os 45b2e7f50SStella Laurenzo 55b2e7f50SStella Laurenzofrom mlir._mlir_libs import get_include_dirs, get_lib_dirs 65b2e7f50SStella Laurenzo 75b2e7f50SStella Laurenzo 85b2e7f50SStella Laurenzoheader_file = os.path.join(get_include_dirs()[0], "mlir-c", "IR.h") 95b2e7f50SStella Laurenzoassert os.path.isfile(header_file), f"Header does not exist: {header_file}" 105b2e7f50SStella Laurenzo 115b2e7f50SStella Laurenzo# Since actual library names are platform specific, just scan the directory 125b2e7f50SStella Laurenzo# for a filename that contains the library name. 135b2e7f50SStella Laurenzoexpected_lib_name = "MLIRPythonCAPI" 145b2e7f50SStella Laurenzoall_libs = os.listdir(get_lib_dirs()[0]) 155b2e7f50SStella Laurenzofound_lib = False 165b2e7f50SStella Laurenzofor file_name in all_libs: 17*f9008e63STobias Hieta if expected_lib_name in file_name: 18*f9008e63STobias Hieta found_lib = True 195b2e7f50SStella Laurenzoassert found_lib, f"Did not find '{expected_lib_name}' lib in {all_libs}" 20