mirror of
https://github.com/osmarks/random-stuff
synced 2024-11-12 23:29:55 +00:00
65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
import tempfile, subprocess, ctypes, os, sys
|
|
|
|
|
|
def c_wrapper(file):
|
|
print("Compiling", file)
|
|
temp = tempfile.mktemp(prefix="lib-compile-")
|
|
print(temp)
|
|
if subprocess.run(["gcc", file, "-o", temp, "-shared", "-fPIC", "-march=native"]).returncode != 0:
|
|
raise ValueError("compilation failed")
|
|
library = ctypes.CDLL(temp)
|
|
entry = library.entry
|
|
entry.restype = ctypes.c_bool
|
|
def wrapper(s1, s2):
|
|
return entry(s1.encode("ascii"), s2.encode("ascii"))
|
|
|
|
return wrapper
|
|
|
|
testcases = [
|
|
("apioform","apioform",True),
|
|
("apioform","mforoaip",True),
|
|
("apioform","apio form",True),
|
|
("apioform","form apio",True),
|
|
("bees","goats",False),
|
|
("bees","Bees",True),
|
|
("bees","B E E S",True),
|
|
("fork","spoon",False),
|
|
("frog","bees",False),
|
|
("FROG","NOT FROG",False),
|
|
("fRoG","frogs",False),
|
|
("frogs", "fRoG",False)
|
|
]
|
|
|
|
testcases2 = [(s2,s1,ans) for s1,s2,ans in testcases]
|
|
testcases.extend(testcases2)
|
|
|
|
def test(entry):
|
|
for s1, s2, res in testcases:
|
|
ans = entry(s1,s2)
|
|
if type(ans) != bool:
|
|
print(f"invalid return type {s1,s2,res,ans}")
|
|
return False
|
|
if ans != res:
|
|
print(f"invalid answer {s1,s2,res,ans}")
|
|
return False
|
|
else:
|
|
print("tests passed successfully")
|
|
return True
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) != 3:
|
|
sys.exit("usage: {sys.argv[0]} py <modulename> or {sys.argv[0]} c <filename>")
|
|
if sys.argv[1] == "py":
|
|
import importlib
|
|
entry = importlib.import_module(sys.argv[2]).entry
|
|
elif sys.argv[1] == "c":
|
|
entry = c_wrapper(sys.argv[2])
|
|
else:
|
|
sys.exit("usage: {sys.argv[0]} py <modulename> or {sys.argv[0]} c <filename>")
|
|
|
|
res = test(entry)
|
|
sys.exit(0 if res else 1)
|
|
|
|
|
|
|