math.big: implement big.integer in V (#11352)

vlib/math/big/array_ops_test.v

@@ -0,0 +1,221 @@
+module big
+
+fn test_add_digit_array_01() {
+	a := [u32(1), 1, 1]
+	b := [u32(1), 1, 1]
+	mut c := []u32{len: 4}
+	add_digit_array(a, b, mut c)
+
+	assert c == [u32(2), 2, 2]
+}
+
+fn test_add_digit_array_02() {
+	a := [u32(1), u32(1) << 31, 1]
+	b := [u32(1), u32(1) << 31, 1]
+	mut c := []u32{len: 4}
+	add_digit_array(a, b, mut c)
+
+	assert c == [u32(2), 0, 3]
+}
+
+fn test_add_digit_array_03() {
+	a := [u32(1), (u32(1) << 31) + u32(34), 1]
+	b := [u32(242), u32(1) << 31, 1]
+	mut c := []u32{len: 4}
+	add_digit_array(a, b, mut c)
+
+	assert c == [u32(243), 34, 3]
+}
+
+fn test_add_digit_array_04() {
+	a := [u32(0)]
+	b := [u32(1), 3, 4]
+	mut c := []u32{len: 4}
+	add_digit_array(a, b, mut c)
+
+	assert c == [u32(1), 3, 4]
+}
+
+fn test_add_digit_array_05() {
+	a := [u32(1), 3, 4]
+	b := [u32(0)]
+	mut c := []u32{len: 4}
+	add_digit_array(a, b, mut c)
+
+	assert c == [u32(1), 3, 4]
+}
+
+fn test_add_digit_array_06() {
+	a := [u32(46), 13, 462, 13]
+	b := [u32(1), 3, 4]
+	mut c := []u32{len: 5}
+	add_digit_array(a, b, mut c)
+
+	assert c == [u32(47), 16, 466, 13]
+}
+
+fn test_subtract_digit_array_01() {
+	a := [u32(2), 2, 2, 2, 2]
+	b := [u32(1), 1, 2, 1, 1]
+	mut c := []u32{len: a.len}
+	subtract_digit_array(a, b, mut c)
+
+	assert c == [u32(1), 1, 0, 1, 1]
+}
+
+fn test_subtract_digit_array_02() {
+	a := [u32(0), 0, 0, 0, 1]
+	b := [u32(0), 0, 1]
+	mut c := []u32{len: a.len}
+	subtract_digit_array(a, b, mut c)
+
+	assert c == [u32(0), 0, u32(-1), u32(-1)]
+}
+
+fn test_subtract_digit_array_03() {
+	a := [u32(0), 0, 0, 0, 1, 13]
+	b := [u32(0), 0, 1]
+	mut c := []u32{len: a.len}
+	subtract_digit_array(a, b, mut c)
+
+	assert c == [u32(0), 0, u32(-1), u32(-1), 0, 13]
+}
+
+fn test_multiply_digit_array_01() {
+	a := [u32(0), 0, 0, 1]
+	b := [u32(0), 0, 1]
+	mut c := []u32{len: a.len + b.len}
+	multiply_digit_array(a, b, mut c)
+
+	assert c == [u32(0), 0, 0, 0, 0, 1]
+}
+
+fn test_multiply_digit_array_02() {
+	a := []u32{len: 0}
+	b := [u32(0), 0, 1]
+	mut c := []u32{len: a.len + b.len}
+	multiply_digit_array(a, b, mut c)
+
+	assert c == []
+
+	c = []u32{len: a.len + b.len}
+	multiply_digit_array(b, a, mut c)
+
+	assert c == []
+}
+
+fn test_compare_digit_array_01() {
+	a := [u32(0), 0, 2]
+	b := [u32(0), 0, 4]
+
+	assert compare_digit_array(a, b) < 0
+	assert compare_digit_array(b, a) > 0
+	assert compare_digit_array(a, a) == 0
+	assert compare_digit_array(b, b) == 0
+}
+
+fn test_compare_digit_array_02() {
+	a := [u32(0), 0, 2324, 0, 124]
+	b := [u32(0), 0, 4, 0, 0, 1]
+
+	assert compare_digit_array(a, b) < 0
+	assert compare_digit_array(b, a) > 0
+	assert compare_digit_array(a, a) == 0
+	assert compare_digit_array(b, b) == 0
+}
+
+fn test_divide_digit_array_01() {
+	a := [u32(14)]
+	b := [u32(2)]
+	mut q := []u32{cap: 1}
+	mut r := []u32{cap: 1}
+
+	divide_digit_array(a, b, mut q, mut r)
+	assert q == [u32(7)]
+	assert r == []u32{len: 0}
+}
+
+fn test_divide_digit_array_02() {
+	a := [u32(14)]
+	b := [u32(15)]
+	mut q := []u32{cap: 1}
+	mut r := []u32{cap: 1}
+
+	divide_digit_array(a, b, mut q, mut r)
+	assert q == []u32{len: 0}
+	assert r == a
+}
+
+fn test_divide_digit_array_03() {
+	a := [u32(0), 4]
+	b := [u32(0), 1]
+	mut q := []u32{cap: a.len - b.len + 1}
+	mut r := []u32{cap: a.len}
+
+	divide_digit_array(a, b, mut q, mut r)
+	assert q == [u32(4)]
+	assert r == []u32{len: 0}
+}
+
+fn test_divide_digit_array_04() {
+	a := [u32(2), 4]
+	b := [u32(0), 1]
+	mut q := []u32{cap: a.len - b.len + 1}
+	mut r := []u32{cap: a.len}
+
+	divide_digit_array(a, b, mut q, mut r)
+	assert q == [u32(4)]
+	assert r == [u32(2)]
+}
+
+fn test_divide_digit_array_05() {
+	a := [u32(3)]
+	b := [u32(2)]
+	mut q := []u32{cap: a.len - b.len + 1}
+	mut r := []u32{cap: a.len}
+
+	divide_digit_array(a, b, mut q, mut r)
+	assert q == [u32(1)]
+	assert r == [u32(1)]
+}
+
+fn test_left_and_right_shift() {
+	a := [u32(1), 1, 1]
+	mut r := [u32(2), 2, 2]
+	mut b := []u32{len: 3, init: 0}
+	shift_digits_left(a, 1, mut b)
+	assert r == b
+	shift_digits_right(r, 1, mut r)
+	assert r == a
+	shift_digits_left(r, 1, mut r)
+	assert r == b
+
+	mut c := [u32(0xffffffff)]
+	shift_digits_left(c, 16, mut c)
+	assert c == [u32(0xfffff0000), 0xffff]
+	shift_digits_right(c, 8, mut c)
+	assert c == [u32(0xfffffff00), 0xff]
+	shift_digits_right(c, 16, mut c)
+	assert c == [u32(0x00ffffff)]
+	shift_digits_right(c, 16, mut c)
+	assert c == [u32(0xff)]
+	shift_digits_right(c, 16, mut c)
+	assert c == []u32{len: 0}
+}
+
+fn test_or_digit_array() {
+	a := [u32(10), 10, 10]
+	b := [u32(5), 5, 5]
+	mut c := []u32{len: 3, init: 0}
+	bitwise_or_digit_array(a, b, mut c)
+	assert c == [u32(15), 15, 15]
+
+	bitwise_or_digit_array(a, a, mut c)
+	assert c == a
+
+	x := [u32(10), 10, 10, 42, 42]
+	y := [u32(2), 2, 5, 2]
+	mut d := []u32{len: 5, init: 0}
+	bitwise_or_digit_array(y, x, mut d)
+	assert d == [u32(10), 10, 15, 42, 42]
+}