mirror of https://github.com/pyodide/pyodide.git
70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
from collections import namedtuple
|
|
from pathlib import Path
|
|
import sys
|
|
|
|
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
|
|
|
|
|
|
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", "ldflags", "host"])
|
|
args = Args(cflags="", ldflags="", host="")
|
|
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 -shared -c test.bc -o test.wasm"
|
|
)
|
|
|
|
# check ldflags injection
|
|
args = Args(cflags="", ldflags="-lm", host="")
|
|
assert (
|
|
handle_command_wrap("gcc -shared -c test.o -o test.so", args)
|
|
== "emcc -lm -shared -c test.bc -o test.wasm"
|
|
)
|
|
|
|
# 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", "ldflags", "host"])
|
|
args = Args(cflags="", ldflags="", host="")
|
|
assert handle_command_wrap(
|
|
"gcc -shared -c test.o -B /compiler_compat -o test.so", args
|
|
) == ("emcc -shared -c test.bc -o test.wasm")
|