76 lines
2.1 KiB
Python
76 lines
2.1 KiB
Python
"""Helpers to check build environment before actual build of scikit-learn"""
|
|
|
|
import glob
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import textwrap
|
|
|
|
from setuptools.command.build_ext import customize_compiler, new_compiler
|
|
|
|
|
|
def compile_test_program(code, extra_preargs=None, extra_postargs=None):
|
|
"""Check that some C code can be compiled and run"""
|
|
ccompiler = new_compiler()
|
|
customize_compiler(ccompiler)
|
|
|
|
start_dir = os.path.abspath(".")
|
|
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
try:
|
|
os.chdir(tmp_dir)
|
|
|
|
# Write test program
|
|
with open("test_program.c", "w") as f:
|
|
f.write(code)
|
|
|
|
os.mkdir("objects")
|
|
|
|
# Compile, test program
|
|
ccompiler.compile(
|
|
["test_program.c"], output_dir="objects", extra_postargs=extra_postargs
|
|
)
|
|
|
|
# Link test program
|
|
objects = glob.glob(os.path.join("objects", "*" + ccompiler.obj_extension))
|
|
ccompiler.link_executable(
|
|
objects,
|
|
"test_program",
|
|
extra_preargs=extra_preargs,
|
|
extra_postargs=extra_postargs,
|
|
)
|
|
|
|
if "PYTHON_CROSSENV" not in os.environ:
|
|
# Run test program if not cross compiling
|
|
# will raise a CalledProcessError if return code was non-zero
|
|
output = subprocess.check_output("./test_program")
|
|
output = output.decode(sys.stdout.encoding or "utf-8").splitlines()
|
|
else:
|
|
# Return an empty output if we are cross compiling
|
|
# as we cannot run the test_program
|
|
output = []
|
|
except Exception:
|
|
raise
|
|
finally:
|
|
os.chdir(start_dir)
|
|
|
|
return output
|
|
|
|
|
|
def basic_check_build():
|
|
"""Check basic compilation and linking of C code"""
|
|
if "PYODIDE" in os.environ:
|
|
# The following check won't work in pyodide
|
|
return
|
|
|
|
code = textwrap.dedent(
|
|
"""\
|
|
#include <stdio.h>
|
|
int main(void) {
|
|
return 0;
|
|
}
|
|
"""
|
|
)
|
|
compile_test_program(code)
|