big: Fix mul.

core/math/big/test.py

@@ -1,5 +1,6 @@
 from  math import *
 from ctypes import *
+from random import *
 import os
 
 #
@@ -38,6 +39,9 @@ except:
 	print("Couldn't find or load " + LIB_PATH + ".")
 	exit(1)
 
+#
+# res = a + b, err
+#
 try:
 	l.test_add_two.argtypes = [c_char_p, c_char_p, c_longlong]
 	l.test_add_two.restype  = Res
@@ -47,6 +51,44 @@ except:
 
 add_two = l.test_add_two
 
+#
+# res = a - b, err
+#
+try:
+	l.test_sub_two.argtypes = [c_char_p, c_char_p, c_longlong]
+	l.test_sub_two.restype  = Res
+except:
+	print("Couldn't find exported function 'test_sub_two'")
+	exit(2)
+
+sub_two = l.test_sub_two
+
+#
+# res = a * b, err
+#
+try:
+	l.test_mul_two.argtypes = [c_char_p, c_char_p, c_longlong]
+	l.test_mul_two.restype  = Res
+except:
+	print("Couldn't find exported function 'test_add_two'")
+	exit(2)
+
+mul_two = l.test_mul_two
+
+#
+# res = a / b, err
+#
+try:
+	l.test_div_two.argtypes = [c_char_p, c_char_p, c_longlong]
+	l.test_div_two.restype  = Res
+except:
+	print("Couldn't find exported function 'test_div_two'")
+	exit(2)
+
+div_two = l.test_div_two
+
+
+
 try:
 	l.test_error_string.argtypes = [c_byte]
 	l.test_error_string.restype  = c_char_p
@@ -91,15 +133,52 @@ def test_add_two(a = 0, b = 0, radix = 10, expected_error = E_None, expected_res
 		expected_result = a + b
 	return test("test_add_two", res, [str(a), str(b), radix], expected_error, expected_result)
 
+def test_sub_two(a = 0, b = 0, radix = 10, expected_error = E_None, expected_result = None):
+	res = sub_two(str(a).encode('utf-8'), str(b).encode('utf-8'), radix)
+	if expected_result == None:
+		expected_result = a - b
+	return test("test_sub_two", res, [str(a), str(b), radix], expected_error, expected_result)
+
+def test_mul_two(a = 0, b = 0, radix = 10, expected_error = E_None, expected_result = None):
+	res = mul_two(str(a).encode('utf-8'), str(b).encode('utf-8'), radix)
+	if expected_result == None:
+		expected_result = a * b
+	return test("test_mul_two", res, [str(a), str(b), radix], expected_error, expected_result)
+
+def test_div_two(a = 0, b = 0, radix = 10, expected_error = E_None, expected_result = None):
+	res = div_two(str(a).encode('utf-8'), str(b).encode('utf-8'), radix)
+	if expected_result == None:
+		expected_result = a // b if b != 0 else None
+	return test("test_add_two", res, [str(a), str(b), radix], expected_error, expected_result)
+
+# TODO(Jeroen): Make sure tests cover edge cases, fast paths, and so on.
+#
+# The last two arguments in tests are the expected error and expected result.
+#
+# The expected error defaults to None.
+# By default the Odin implementation will be tested against the Python one.
+# You can override that by supplying an expected result as the last argument instead.
 
 TESTS = {
 	test_add_two: [
-		[ 1234, 5432,  10,                     ],
-		[ 1234, 5432, 110, E_Invalid_Argument, ],
+		[ 1234,   5432,    10, ],
+		[ 1234,   5432,   110, E_Invalid_Argument, ],
+	],
+	test_sub_two: [
+		[ 1234,   5432,    10, ],
+	],
+	test_mul_two: [
+		[ 1234,   5432,    10, ],
+		[ 1099243943008198766717263669950239669, 137638828577110581150675834234248871, 10, ]
+	],
+	test_div_two: [
+		[ 54321, 12345,    10, ],
+		[ 55431,     0,    10, E_Division_by_Zero, ],
 	],
 }
 
 if __name__ == '__main__':
+	print()
 	print("---- core:math/big tests ----")
 	print()
 
@@ -112,4 +191,27 @@ if __name__ == '__main__':
 			else:
 				count_fail += 1
 
-		print("{}: {} passes, {} failures.".format(test_proc.__name__, count_pass, count_fail))
+		print("{}: {} passes, {} failures.".format(test_proc.__name__, count_pass, count_fail))
+
+	print()		
+	print("---- core:math/big random tests ----")
+	print()
+
+	for test_proc in [test_add_two, test_sub_two, test_mul_two, test_div_two]:
+		count_pass = 0
+		count_fail = 0
+
+		a = randint(0, 1 << 120)
+		b = randint(0, 1 << 120)
+		res = None
+
+		# We've already tested division by zero above.
+		if b == 0 and test_proc == test_div_two:
+			b = b + 1
+
+		if test_proc(a, b):
+			count_pass += 1
+		else:
+			count_fail += 1
+
+		print("{} random: {} passes, {} failures.".format(test_proc.__name__, count_pass, count_fail))