big: Start test suite.

core/math/big/test.py

@@ -0,0 +1,79 @@
+from  math import *
+from ctypes import *
+import os
+
+#
+# Where is the DLL? If missing, build using: `odin build . -build-mode:dll`
+#
+LIB_PATH = os.getcwd() + os.sep + "big.dll"
+
+#
+# Result values will be passed in a struct { res: cstring, err: Error }
+#
+class Res(Structure):
+	_fields_ = [("res", c_char_p), ("err", c_byte)]
+
+#
+# Error enum values
+#
+E_None                   = 0
+E_Out_Of_Memory          = 1
+E_Invalid_Pointer        = 2
+E_Invalid_Argument       = 3
+E_Unknown_Error          = 4
+E_Max_Iterations_Reached = 5
+E_Buffer_Overflow        = 6
+E_Integer_Overflow       = 7
+E_Division_by_Zero       = 8
+E_Math_Domain_Error      = 9
+E_Unimplemented          = 127
+
+#
+# Set up exported procedures
+#
+
+try:
+	l = cdll.LoadLibrary(LIB_PATH)
+except:
+	print("Couldn't find or load " + LIB_PATH + ".")
+	exit(1)
+
+try:
+	l.test_add_two.argtypes = [c_char_p, c_char_p, c_longlong]
+	l.test_add_two.restype  = Res
+except:
+	print("Couldn't find exported function 'test_add_two'")
+	exit(2)
+
+add_two = l.test_add_two
+
+try:
+	l.test_error_string.argtypes = [c_byte]
+	l.test_error_string.restype  = c_char_p
+except:
+	print("Couldn't find exported function 'test_error_string'")
+	exit(2)
+
+def error(res: Res, param=[]):
+	if res.err != E_None:
+		error_type = l.test_error_string(res.err).decode('utf-8')
+		error_loc  = res.res.decode('utf-8')
+
+		error_string = "'{}' error in '{}'".format(error_type, error_loc)
+		if len(param):
+			error_string += " with params {}".format(param)
+
+		print(error_string, flush=True)
+		os._exit(res.err)
+
+
+def test_add_two(a = 0, b = 0, radix = 10):
+	res = add_two(str(a).encode('utf-8'), str(b).encode('utf-8'), radix)
+	error(res, [str(a), str(b), radix])
+
+if __name__ == '__main__':
+	print("---- core:math/big tests ----")
+	print()
+
+	test_add_two(1234, 5432, 10)
+	test_add_two(1234, 5432, 110)