mirror of https://github.com/pyodide/pyodide.git
128 lines
3.8 KiB
Python
128 lines
3.8 KiB
Python
from collections import namedtuple
|
|
from pathlib import Path
|
|
import sys
|
|
import argparse
|
|
|
|
sys.path.append(str(Path(__file__).parents[2]))
|
|
|
|
from pyodide_build.pywasmcross import handle_command # noqa: E402
|
|
from pyodide_build.pywasmcross import f2c # noqa: E402
|
|
from pyodide_build.pywasmcross import make_parser
|
|
|
|
|
|
def _args_wrapper(func):
|
|
"""Convert function to take as input / return a string instead of a
|
|
list of arguments
|
|
|
|
Also sets dryrun=True
|
|
"""
|
|
|
|
def _inner(line, *pargs):
|
|
args = line.split()
|
|
res = func(args, *pargs, dryrun=True)
|
|
if hasattr(res, "__len__"):
|
|
return " ".join(res)
|
|
else:
|
|
return res
|
|
|
|
return _inner
|
|
|
|
|
|
handle_command_wrap = _args_wrapper(handle_command)
|
|
f2c_wrap = _args_wrapper(f2c)
|
|
|
|
|
|
def test_handle_command():
|
|
Args = namedtuple(
|
|
"args", ["cflags", "cxxflags", "ldflags", "host", "replace_libs", "install_dir"]
|
|
)
|
|
args = Args(
|
|
cflags="", cxxflags="", ldflags="", host="", replace_libs="", install_dir=""
|
|
)
|
|
assert handle_command_wrap("gcc -print-multiarch", args) is None
|
|
assert handle_command_wrap("gcc test.c", args) == "emcc test.c"
|
|
assert (
|
|
handle_command_wrap("gcc -shared -c test.o -o test.so", args)
|
|
== "emcc -c test.o -o test.so"
|
|
)
|
|
|
|
# check cxxflags injection and cpp detection
|
|
args = Args(
|
|
cflags="-I./lib2",
|
|
cxxflags="-std=c++11",
|
|
ldflags="-lm",
|
|
host="",
|
|
replace_libs="",
|
|
install_dir="",
|
|
)
|
|
assert (
|
|
handle_command_wrap("gcc -I./lib1 test.cpp -o test.o", args)
|
|
== "em++ -I./lib2 -std=c++11 -I./lib1 test.cpp -o test.o"
|
|
)
|
|
|
|
# check ldflags injection
|
|
args = Args(
|
|
cflags="", cxxflags="", ldflags="-lm", host="", replace_libs="", install_dir=""
|
|
)
|
|
assert (
|
|
handle_command_wrap("gcc -shared -c test.o -o test.so", args)
|
|
== "emcc -lm -c test.o -o test.so"
|
|
)
|
|
|
|
# check library replacement and removal of double libraries
|
|
args = Args(
|
|
cflags="",
|
|
cxxflags="",
|
|
ldflags="",
|
|
host="",
|
|
replace_libs="bob=fred",
|
|
install_dir="",
|
|
)
|
|
assert (
|
|
handle_command_wrap("gcc -shared test.o -lbob -ljim -ljim -o test.so", args)
|
|
== "emcc test.o -lfred -ljim -o test.so"
|
|
)
|
|
|
|
# compilation checks in numpy
|
|
assert handle_command_wrap("gcc /usr/file.c", args) is None
|
|
|
|
|
|
def test_f2c():
|
|
assert f2c_wrap("gfortran test.f") == "gfortran test.c"
|
|
assert f2c_wrap("gcc test.c") is None
|
|
assert f2c_wrap("gfortran --version") is None
|
|
assert (
|
|
f2c_wrap("gfortran --shared -c test.o -o test.so")
|
|
== "gfortran --shared -c test.o -o test.so"
|
|
)
|
|
|
|
|
|
def test_conda_compiler_compat():
|
|
Args = namedtuple(
|
|
"args", ["cflags", "cxxflags", "ldflags", "host", "replace_libs", "install_dir"]
|
|
)
|
|
args = Args(
|
|
cflags="", cxxflags="", ldflags="", host="", replace_libs="", install_dir=""
|
|
)
|
|
assert handle_command_wrap(
|
|
"gcc -shared -c test.o -B /compiler_compat -o test.so", args
|
|
) == ("emcc -c test.o -o test.so")
|
|
|
|
|
|
def test_environment_var_substitution(monkeypatch):
|
|
monkeypatch.setenv("PYODIDE_BASE", "pyodide_build_dir")
|
|
monkeypatch.setenv("BOB", "Robert Mc Roberts")
|
|
monkeypatch.setenv("FRED", "Frederick F. Freddertson Esq.")
|
|
monkeypatch.setenv("JIM", "James Ignatius Morrison:Jimmy")
|
|
call_args = 'pywasmcross.py --ldflags "-l$(PYODIDE_BASE)" --cxxflags $(BOB) --cflags $(FRED) --replace-libs $(JIM)'
|
|
monkeypatch.setattr(sys, "argv", call_args.split(" "))
|
|
parser = argparse.ArgumentParser()
|
|
make_parser(parser)
|
|
args = parser.parse_args()
|
|
assert (
|
|
args.cflags == "Frederick F. Freddertson Esq."
|
|
and args.cxxflags == "Robert Mc Roberts"
|
|
and args.ldflags == '"-lpyodide_build_dir"'
|
|
and args.replace_libs == "James Ignatius Morrison:Jimmy"
|
|
)
|