aboutsummaryrefslogtreecommitdiff
path: root/core/math
diff options
context:
space:
mode:
authorJeroen van Rijn <Kelimion@users.noreply.github.com>2021-09-07 12:06:58 +0200
committerJeroen van Rijn <Kelimion@users.noreply.github.com>2021-09-07 12:06:58 +0200
commit008048e199bdbf6d84fad756a774d4e5e2420edc (patch)
tree6efdc4445e320018c0287088882d05909e4e48fa /core/math
parent48bfce2efcf99d2b79bdbb19ae451973a757d006 (diff)
big: Move tests into their own package.
Diffstat (limited to 'core/math')
-rw-r--r--core/math/big/build.bat11
-rw-r--r--core/math/big/test.odin389
-rw-r--r--core/math/big/tests/build.bat8
-rw-r--r--core/math/big/tests/test.odin390
-rw-r--r--core/math/big/tests/test.py (renamed from core/math/big/test.py)6
5 files changed, 401 insertions, 403 deletions
diff --git a/core/math/big/build.bat b/core/math/big/build.bat
deleted file mode 100644
index 308f45d31..000000000
--- a/core/math/big/build.bat
+++ /dev/null
@@ -1,11 +0,0 @@
-@echo off
-:odin run . -vet -strict-style
-: -define:MATH_BIG_USE_FROBENIUS_TEST=true
-
-set TEST_ARGS=-fast-tests
-:set TEST_ARGS=
-:odin build . -build-mode:shared -show-timings -o:minimal -no-bounds-check -define:MATH_BIG_EXE=false && python test.py %TEST_ARGS%
-:odin build . -build-mode:shared -show-timings -o:size -no-bounds-check -define:MATH_BIG_EXE=false && python test.py %TEST_ARGS%
-:odin build . -build-mode:shared -show-timings -o:size -define:MATH_BIG_EXE=false && python test.py %TEST_ARGS%
-:odin build . -build-mode:shared -show-timings -o:speed -no-bounds-check -define:MATH_BIG_EXE=false && python test.py %TEST_ARGS%
-:odin build . -build-mode:shared -show-timings -o:speed -define:MATH_BIG_EXE=false && python test.py -fast-tests %TEST_ARGS% \ No newline at end of file
diff --git a/core/math/big/test.odin b/core/math/big/test.odin
deleted file mode 100644
index 8db2ab7eb..000000000
--- a/core/math/big/test.odin
+++ /dev/null
@@ -1,389 +0,0 @@
-//+ignore
-/*
- Copyright 2021 Jeroen van Rijn <nom@duclavier.com>.
- Made available under Odin's BSD-3 license.
-
- An arbitrary precision mathematics implementation in Odin.
- For the theoretical underpinnings, see Knuth's The Art of Computer Programming, Volume 2, section 4.3.
- The code started out as an idiomatic source port of libTomMath, which is in the public domain, with thanks.
-
- This file exports procedures for use with the test.py test suite.
-*/
-package math_big
-
-/*
- TODO: Write tests for `internal_*` and test reusing parameters with the public implementations.
-*/
-
-import "core:runtime"
-import "core:strings"
-
-PyRes :: struct {
- res: cstring,
- err: Error,
-}
-
-@export test_initialize_constants :: proc "c" () -> (res: u64) {
- context = runtime.default_context()
- res = u64(initialize_constants())
- //assert(MUL_KARATSUBA_CUTOFF >= 40);
- return res
-}
-
-@export test_error_string :: proc "c" (err: Error) -> (res: cstring) {
- context = runtime.default_context()
- es := Error_String
- return strings.clone_to_cstring(es[err], context.temp_allocator)
-}
-
-@export test_add :: proc "c" (a, b: cstring) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- aa, bb, sum := &Int{}, &Int{}, &Int{}
- defer internal_destroy(aa, bb, sum)
-
- if err = atoi(aa, string(a), 16); err != nil { return PyRes{res=":add:atoi(a):", err=err} }
- if err = atoi(bb, string(b), 16); err != nil { return PyRes{res=":add:atoi(b):", err=err} }
- if bb.used == 1 {
- if err = #force_inline internal_add(sum, aa, bb.digit[0]); err != nil { return PyRes{res=":add:add(sum,a,b):", err=err} }
- } else {
- if err = #force_inline internal_add(sum, aa, bb); err != nil { return PyRes{res=":add:add(sum,a,b):", err=err} }
- }
-
- r: cstring
- r, err = int_itoa_cstring(sum, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":add:itoa(sum):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-@export test_sub :: proc "c" (a, b: cstring) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- aa, bb, sum := &Int{}, &Int{}, &Int{}
- defer internal_destroy(aa, bb, sum)
-
- if err = atoi(aa, string(a), 16); err != nil { return PyRes{res=":sub:atoi(a):", err=err} }
- if err = atoi(bb, string(b), 16); err != nil { return PyRes{res=":sub:atoi(b):", err=err} }
- if bb.used == 1 {
- if err = #force_inline internal_sub(sum, aa, bb.digit[0]); err != nil { return PyRes{res=":sub:sub(sum,a,b):", err=err} }
- } else {
- if err = #force_inline internal_sub(sum, aa, bb); err != nil { return PyRes{res=":sub:sub(sum,a,b):", err=err} }
- }
-
- r: cstring
- r, err = int_itoa_cstring(sum, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":sub:itoa(sum):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-@export test_mul :: proc "c" (a, b: cstring) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- aa, bb, product := &Int{}, &Int{}, &Int{}
- defer internal_destroy(aa, bb, product)
-
- if err = atoi(aa, string(a), 16); err != nil { return PyRes{res=":mul:atoi(a):", err=err} }
- if err = atoi(bb, string(b), 16); err != nil { return PyRes{res=":mul:atoi(b):", err=err} }
- if err = #force_inline internal_mul(product, aa, bb); err != nil { return PyRes{res=":mul:mul(product,a,b):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(product, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":mul:itoa(product):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-@export test_sqr :: proc "c" (a: cstring) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- aa, square := &Int{}, &Int{}
- defer internal_destroy(aa, square)
-
- if err = atoi(aa, string(a), 16); err != nil { return PyRes{res=":sqr:atoi(a):", err=err} }
- if err = #force_inline internal_sqr(square, aa); err != nil { return PyRes{res=":sqr:sqr(square,a):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(square, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":sqr:itoa(square):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-/*
- NOTE(Jeroen): For simplicity, we don't return the quotient and the remainder, just the quotient.
-*/
-@export test_div :: proc "c" (a, b: cstring) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- aa, bb, quotient := &Int{}, &Int{}, &Int{}
- defer internal_destroy(aa, bb, quotient)
-
- if err = atoi(aa, string(a), 16); err != nil { return PyRes{res=":div:atoi(a):", err=err} }
- if err = atoi(bb, string(b), 16); err != nil { return PyRes{res=":div:atoi(b):", err=err} }
- if err = #force_inline internal_div(quotient, aa, bb); err != nil { return PyRes{res=":div:div(quotient,a,b):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(quotient, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":div:itoa(quotient):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-
-/*
- res = log(a, base)
-*/
-@export test_log :: proc "c" (a: cstring, base := DIGIT(2)) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
- l: int
-
- aa := &Int{}
- defer internal_destroy(aa)
-
- if err = atoi(aa, string(a), 16); err != nil { return PyRes{res=":log:atoi(a):", err=err} }
- if l, err = #force_inline internal_log(aa, base); err != nil { return PyRes{res=":log:log(a, base):", err=err} }
-
- #force_inline internal_zero(aa)
- aa.digit[0] = DIGIT(l) & _MASK
- aa.digit[1] = DIGIT(l) >> _DIGIT_BITS
- aa.used = 2
- clamp(aa)
-
- r: cstring
- r, err = int_itoa_cstring(aa, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":log:itoa(res):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-/*
- dest = base^power
-*/
-@export test_pow :: proc "c" (base: cstring, power := int(2)) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- dest, bb := &Int{}, &Int{}
- defer internal_destroy(dest, bb)
-
- if err = atoi(bb, string(base), 16); err != nil { return PyRes{res=":pow:atoi(base):", err=err} }
- if err = #force_inline internal_pow(dest, bb, power); err != nil { return PyRes{res=":pow:pow(dest, base, power):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(dest, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":log:itoa(res):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-/*
- dest = sqrt(src)
-*/
-@export test_sqrt :: proc "c" (source: cstring) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- src := &Int{}
- defer internal_destroy(src)
-
- if err = atoi(src, string(source), 16); err != nil { return PyRes{res=":sqrt:atoi(src):", err=err} }
- if err = #force_inline internal_sqrt(src, src); err != nil { return PyRes{res=":sqrt:sqrt(src):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(src, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":log:itoa(res):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-/*
- dest = root_n(src, power)
-*/
-@export test_root_n :: proc "c" (source: cstring, power: int) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- src := &Int{}
- defer internal_destroy(src)
-
- if err = atoi(src, string(source), 16); err != nil { return PyRes{res=":root_n:atoi(src):", err=err} }
- if err = #force_inline internal_root_n(src, src, power); err != nil { return PyRes{res=":root_n:root_n(src):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(src, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":root_n:itoa(res):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-/*
- dest = shr_digit(src, digits)
-*/
-@export test_shr_digit :: proc "c" (source: cstring, digits: int) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- src := &Int{}
- defer internal_destroy(src)
-
- if err = atoi(src, string(source), 16); err != nil { return PyRes{res=":shr_digit:atoi(src):", err=err} }
- if err = #force_inline internal_shr_digit(src, digits); err != nil { return PyRes{res=":shr_digit:shr_digit(src):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(src, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":shr_digit:itoa(res):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-/*
- dest = shl_digit(src, digits)
-*/
-@export test_shl_digit :: proc "c" (source: cstring, digits: int) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- src := &Int{}
- defer internal_destroy(src)
-
- if err = atoi(src, string(source), 16); err != nil { return PyRes{res=":shl_digit:atoi(src):", err=err} }
- if err = #force_inline internal_shl_digit(src, digits); err != nil { return PyRes{res=":shl_digit:shr_digit(src):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(src, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":shl_digit:itoa(res):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-/*
- dest = shr(src, bits)
-*/
-@export test_shr :: proc "c" (source: cstring, bits: int) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- src := &Int{}
- defer internal_destroy(src)
-
- if err = atoi(src, string(source), 16); err != nil { return PyRes{res=":shr:atoi(src):", err=err} }
- if err = #force_inline internal_shr(src, src, bits); err != nil { return PyRes{res=":shr:shr(src, bits):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(src, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":shr:itoa(res):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-/*
- dest = shr_signed(src, bits)
-*/
-@export test_shr_signed :: proc "c" (source: cstring, bits: int) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- src := &Int{}
- defer internal_destroy(src)
-
- if err = atoi(src, string(source), 16); err != nil { return PyRes{res=":shr_signed:atoi(src):", err=err} }
- if err = #force_inline internal_shr_signed(src, src, bits); err != nil { return PyRes{res=":shr_signed:shr_signed(src, bits):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(src, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":shr_signed:itoa(res):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-/*
- dest = shl(src, bits)
-*/
-@export test_shl :: proc "c" (source: cstring, bits: int) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- src := &Int{}
- defer internal_destroy(src)
-
- if err = atoi(src, string(source), 16); err != nil { return PyRes{res=":shl:atoi(src):", err=err} }
- if err = #force_inline internal_shl(src, src, bits); err != nil { return PyRes{res=":shl:shl(src, bits):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(src, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":shl:itoa(res):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-/*
- dest = factorial(n)
-*/
-@export test_factorial :: proc "c" (n: int) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- dest := &Int{}
- defer internal_destroy(dest)
-
- if err = #force_inline internal_int_factorial(dest, n); err != nil { return PyRes{res=":factorial:factorial(n):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(dest, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":factorial:itoa(res):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-/*
- dest = gcd(a, b)
-*/
-@export test_gcd :: proc "c" (a, b: cstring) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- ai, bi, dest := &Int{}, &Int{}, &Int{}
- defer internal_destroy(ai, bi, dest)
-
- if err = atoi(ai, string(a), 16); err != nil { return PyRes{res=":gcd:atoi(a):", err=err} }
- if err = atoi(bi, string(b), 16); err != nil { return PyRes{res=":gcd:atoi(b):", err=err} }
- if err = #force_inline internal_int_gcd_lcm(dest, nil, ai, bi); err != nil { return PyRes{res=":gcd:gcd(a, b):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(dest, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":gcd:itoa(res):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-/*
- dest = lcm(a, b)
-*/
-@export test_lcm :: proc "c" (a, b: cstring) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
-
- ai, bi, dest := &Int{}, &Int{}, &Int{}
- defer internal_destroy(ai, bi, dest)
-
- if err = atoi(ai, string(a), 16); err != nil { return PyRes{res=":lcm:atoi(a):", err=err} }
- if err = atoi(bi, string(b), 16); err != nil { return PyRes{res=":lcm:atoi(b):", err=err} }
- if err = #force_inline internal_int_gcd_lcm(nil, dest, ai, bi); err != nil { return PyRes{res=":lcm:lcm(a, b):", err=err} }
-
- r: cstring
- r, err = int_itoa_cstring(dest, 16, context.temp_allocator)
- if err != nil { return PyRes{res=":lcm:itoa(res):", err=err} }
- return PyRes{res = r, err = nil}
-}
-
-/*
- dest = lcm(a, b)
-*/
-@export test_is_square :: proc "c" (a: cstring) -> (res: PyRes) {
- context = runtime.default_context()
- err: Error
- square: bool
-
- ai := &Int{}
- defer internal_destroy(ai)
-
- if err = atoi(ai, string(a), 16); err != nil { return PyRes{res=":is_square:atoi(a):", err=err} }
- if square, err = #force_inline internal_int_is_square(ai); err != nil { return PyRes{res=":is_square:is_square(a):", err=err} }
-
- if square {
- return PyRes{"True", nil}
- }
- return PyRes{"False", nil}
-} \ No newline at end of file
diff --git a/core/math/big/tests/build.bat b/core/math/big/tests/build.bat
new file mode 100644
index 000000000..6a82106db
--- /dev/null
+++ b/core/math/big/tests/build.bat
@@ -0,0 +1,8 @@
+@echo off
+set TEST_ARGS=-fast-tests
+set TEST_ARGS=
+set OUT_NAME=test_library
+set COMMON=-build-mode:shared -show-timings -no-bounds-check -define:MATH_BIG_EXE=false -vet -strict-style
+:odin build . %COMMON% -o:minimal -out:%OUT_NAME% && python test.py %TEST_ARGS%
+odin build . %COMMON% -o:size -out:%OUT_NAME% && python test.py %TEST_ARGS%
+:odin build . %COMMON% -o:speed -out:%OUT_NAME% && python test.py %TEST_ARGS% \ No newline at end of file
diff --git a/core/math/big/tests/test.odin b/core/math/big/tests/test.odin
new file mode 100644
index 000000000..baf7b3177
--- /dev/null
+++ b/core/math/big/tests/test.odin
@@ -0,0 +1,390 @@
+//+ignore
+/*
+ Copyright 2021 Jeroen van Rijn <nom@duclavier.com>.
+ Made available under Odin's BSD-3 license.
+
+ An arbitrary precision mathematics implementation in Odin.
+ For the theoretical underpinnings, see Knuth's The Art of Computer Programming, Volume 2, section 4.3.
+ The code started out as an idiomatic source port of libTomMath, which is in the public domain, with thanks.
+
+ This file exports procedures for use with the test.py test suite.
+*/
+package math_big_tests
+
+/*
+ TODO: Write tests for `internal_*` and test reusing parameters with the public implementations.
+*/
+
+import "core:runtime"
+import "core:strings"
+import "core:math/big"
+
+PyRes :: struct {
+ res: cstring,
+ err: big.Error,
+}
+
+@export test_initialize_constants :: proc "c" () -> (res: u64) {
+ context = runtime.default_context()
+ res = u64(big.initialize_constants())
+ //assert(MUL_KARATSUBA_CUTOFF >= 40);
+ return res
+}
+
+@export test_error_string :: proc "c" (err: big.Error) -> (res: cstring) {
+ context = runtime.default_context()
+ es := big.Error_String
+ return strings.clone_to_cstring(es[err], context.temp_allocator)
+}
+
+@export test_add :: proc "c" (a, b: cstring) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ aa, bb, sum := &big.Int{}, &big.Int{}, &big.Int{}
+ defer big.internal_destroy(aa, bb, sum)
+
+ if err = big.atoi(aa, string(a), 16); err != nil { return PyRes{res=":add:atoi(a):", err=err} }
+ if err = big.atoi(bb, string(b), 16); err != nil { return PyRes{res=":add:atoi(b):", err=err} }
+ if bb.used == 1 {
+ if err = #force_inline big.internal_add(sum, aa, bb.digit[0]); err != nil { return PyRes{res=":add:add(sum,a,b):", err=err} }
+ } else {
+ if err = #force_inline big.internal_add(sum, aa, bb); err != nil { return PyRes{res=":add:add(sum,a,b):", err=err} }
+ }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(sum, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":add:itoa(sum):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+@export test_sub :: proc "c" (a, b: cstring) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ aa, bb, sum := &big.Int{}, &big.Int{}, &big.Int{}
+ defer big.internal_destroy(aa, bb, sum)
+
+ if err = big.atoi(aa, string(a), 16); err != nil { return PyRes{res=":sub:atoi(a):", err=err} }
+ if err = big.atoi(bb, string(b), 16); err != nil { return PyRes{res=":sub:atoi(b):", err=err} }
+ if bb.used == 1 {
+ if err = #force_inline big.internal_sub(sum, aa, bb.digit[0]); err != nil { return PyRes{res=":sub:sub(sum,a,b):", err=err} }
+ } else {
+ if err = #force_inline big.internal_sub(sum, aa, bb); err != nil { return PyRes{res=":sub:sub(sum,a,b):", err=err} }
+ }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(sum, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":sub:itoa(sum):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+@export test_mul :: proc "c" (a, b: cstring) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ aa, bb, product := &big.Int{}, &big.Int{}, &big.Int{}
+ defer big.internal_destroy(aa, bb, product)
+
+ if err = big.atoi(aa, string(a), 16); err != nil { return PyRes{res=":mul:atoi(a):", err=err} }
+ if err = big.atoi(bb, string(b), 16); err != nil { return PyRes{res=":mul:atoi(b):", err=err} }
+ if err = #force_inline big.internal_mul(product, aa, bb); err != nil { return PyRes{res=":mul:mul(product,a,b):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(product, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":mul:itoa(product):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+@export test_sqr :: proc "c" (a: cstring) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ aa, square := &big.Int{}, &big.Int{}
+ defer big.internal_destroy(aa, square)
+
+ if err = big.atoi(aa, string(a), 16); err != nil { return PyRes{res=":sqr:atoi(a):", err=err} }
+ if err = #force_inline big.internal_sqr(square, aa); err != nil { return PyRes{res=":sqr:sqr(square,a):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(square, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":sqr:itoa(square):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+/*
+ NOTE(Jeroen): For simplicity, we don't return the quotient and the remainder, just the quotient.
+*/
+@export test_div :: proc "c" (a, b: cstring) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ aa, bb, quotient := &big.Int{}, &big.Int{}, &big.Int{}
+ defer big.internal_destroy(aa, bb, quotient)
+
+ if err = big.atoi(aa, string(a), 16); err != nil { return PyRes{res=":div:atoi(a):", err=err} }
+ if err = big.atoi(bb, string(b), 16); err != nil { return PyRes{res=":div:atoi(b):", err=err} }
+ if err = #force_inline big.internal_div(quotient, aa, bb); err != nil { return PyRes{res=":div:div(quotient,a,b):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(quotient, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":div:itoa(quotient):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+
+/*
+ res = log(a, base)
+*/
+@export test_log :: proc "c" (a: cstring, base := big.DIGIT(2)) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+ l: int
+
+ aa := &big.Int{}
+ defer big.internal_destroy(aa)
+
+ if err = big.atoi(aa, string(a), 16); err != nil { return PyRes{res=":log:atoi(a):", err=err} }
+ if l, err = #force_inline big.internal_log(aa, base); err != nil { return PyRes{res=":log:log(a, base):", err=err} }
+
+ #force_inline big.internal_zero(aa)
+ aa.digit[0] = big.DIGIT(l) & big._MASK
+ aa.digit[1] = big.DIGIT(l) >> big._DIGIT_BITS
+ aa.used = 2
+ big.clamp(aa)
+
+ r: cstring
+ r, err = big.int_itoa_cstring(aa, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":log:itoa(res):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+/*
+ dest = base^power
+*/
+@export test_pow :: proc "c" (base: cstring, power := int(2)) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ dest, bb := &big.Int{}, &big.Int{}
+ defer big.internal_destroy(dest, bb)
+
+ if err = big.atoi(bb, string(base), 16); err != nil { return PyRes{res=":pow:atoi(base):", err=err} }
+ if err = #force_inline big.internal_pow(dest, bb, power); err != nil { return PyRes{res=":pow:pow(dest, base, power):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(dest, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":log:itoa(res):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+/*
+ dest = sqrt(src)
+*/
+@export test_sqrt :: proc "c" (source: cstring) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ src := &big.Int{}
+ defer big.internal_destroy(src)
+
+ if err = big.atoi(src, string(source), 16); err != nil { return PyRes{res=":sqrt:atoi(src):", err=err} }
+ if err = #force_inline big.internal_sqrt(src, src); err != nil { return PyRes{res=":sqrt:sqrt(src):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(src, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":log:itoa(res):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+/*
+ dest = root_n(src, power)
+*/
+@export test_root_n :: proc "c" (source: cstring, power: int) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ src := &big.Int{}
+ defer big.internal_destroy(src)
+
+ if err = big.atoi(src, string(source), 16); err != nil { return PyRes{res=":root_n:atoi(src):", err=err} }
+ if err = #force_inline big.internal_root_n(src, src, power); err != nil { return PyRes{res=":root_n:root_n(src):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(src, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":root_n:itoa(res):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+/*
+ dest = shr_digit(src, digits)
+*/
+@export test_shr_digit :: proc "c" (source: cstring, digits: int) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ src := &big.Int{}
+ defer big.internal_destroy(src)
+
+ if err = big.atoi(src, string(source), 16); err != nil { return PyRes{res=":shr_digit:atoi(src):", err=err} }
+ if err = #force_inline big.internal_shr_digit(src, digits); err != nil { return PyRes{res=":shr_digit:shr_digit(src):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(src, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":shr_digit:itoa(res):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+/*
+ dest = shl_digit(src, digits)
+*/
+@export test_shl_digit :: proc "c" (source: cstring, digits: int) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ src := &big.Int{}
+ defer big.internal_destroy(src)
+
+ if err = big.atoi(src, string(source), 16); err != nil { return PyRes{res=":shl_digit:atoi(src):", err=err} }
+ if err = #force_inline big.internal_shl_digit(src, digits); err != nil { return PyRes{res=":shl_digit:shr_digit(src):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(src, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":shl_digit:itoa(res):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+/*
+ dest = shr(src, bits)
+*/
+@export test_shr :: proc "c" (source: cstring, bits: int) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ src := &big.Int{}
+ defer big.internal_destroy(src)
+
+ if err = big.atoi(src, string(source), 16); err != nil { return PyRes{res=":shr:atoi(src):", err=err} }
+ if err = #force_inline big.internal_shr(src, src, bits); err != nil { return PyRes{res=":shr:shr(src, bits):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(src, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":shr:itoa(res):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+/*
+ dest = shr_signed(src, bits)
+*/
+@export test_shr_signed :: proc "c" (source: cstring, bits: int) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ src := &big.Int{}
+ defer big.internal_destroy(src)
+
+ if err = big.atoi(src, string(source), 16); err != nil { return PyRes{res=":shr_signed:atoi(src):", err=err} }
+ if err = #force_inline big.internal_shr_signed(src, src, bits); err != nil { return PyRes{res=":shr_signed:shr_signed(src, bits):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(src, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":shr_signed:itoa(res):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+/*
+ dest = shl(src, bits)
+*/
+@export test_shl :: proc "c" (source: cstring, bits: int) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ src := &big.Int{}
+ defer big.internal_destroy(src)
+
+ if err = big.atoi(src, string(source), 16); err != nil { return PyRes{res=":shl:atoi(src):", err=err} }
+ if err = #force_inline big.internal_shl(src, src, bits); err != nil { return PyRes{res=":shl:shl(src, bits):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(src, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":shl:itoa(res):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+/*
+ dest = factorial(n)
+*/
+@export test_factorial :: proc "c" (n: int) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ dest := &big.Int{}
+ defer big.internal_destroy(dest)
+
+ if err = #force_inline big.internal_int_factorial(dest, n); err != nil { return PyRes{res=":factorial:factorial(n):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(dest, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":factorial:itoa(res):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+/*
+ dest = gcd(a, b)
+*/
+@export test_gcd :: proc "c" (a, b: cstring) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ ai, bi, dest := &big.Int{}, &big.Int{}, &big.Int{}
+ defer big.internal_destroy(ai, bi, dest)
+
+ if err = big.atoi(ai, string(a), 16); err != nil { return PyRes{res=":gcd:atoi(a):", err=err} }
+ if err = big.atoi(bi, string(b), 16); err != nil { return PyRes{res=":gcd:atoi(b):", err=err} }
+ if err = #force_inline big.internal_int_gcd_lcm(dest, nil, ai, bi); err != nil { return PyRes{res=":gcd:gcd(a, b):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(dest, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":gcd:itoa(res):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+/*
+ dest = lcm(a, b)
+*/
+@export test_lcm :: proc "c" (a, b: cstring) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+
+ ai, bi, dest := &big.Int{}, &big.Int{}, &big.Int{}
+ defer big.internal_destroy(ai, bi, dest)
+
+ if err = big.atoi(ai, string(a), 16); err != nil { return PyRes{res=":lcm:atoi(a):", err=err} }
+ if err = big.atoi(bi, string(b), 16); err != nil { return PyRes{res=":lcm:atoi(b):", err=err} }
+ if err = #force_inline big.internal_int_gcd_lcm(nil, dest, ai, bi); err != nil { return PyRes{res=":lcm:lcm(a, b):", err=err} }
+
+ r: cstring
+ r, err = big.int_itoa_cstring(dest, 16, context.temp_allocator)
+ if err != nil { return PyRes{res=":lcm:itoa(res):", err=err} }
+ return PyRes{res = r, err = nil}
+}
+
+/*
+ dest = lcm(a, b)
+*/
+@export test_is_square :: proc "c" (a: cstring) -> (res: PyRes) {
+ context = runtime.default_context()
+ err: big.Error
+ square: bool
+
+ ai := &big.Int{}
+ defer big.internal_destroy(ai)
+
+ if err = big.atoi(ai, string(a), 16); err != nil { return PyRes{res=":is_square:atoi(a):", err=err} }
+ if square, err = #force_inline big.internal_int_is_square(ai); err != nil { return PyRes{res=":is_square:is_square(a):", err=err} }
+
+ if square {
+ return PyRes{"True", nil}
+ }
+ return PyRes{"False", nil}
+} \ No newline at end of file
diff --git a/core/math/big/test.py b/core/math/big/tests/test.py
index 4d314401f..0c346a9bf 100644
--- a/core/math/big/test.py
+++ b/core/math/big/tests/test.py
@@ -101,11 +101,11 @@ if args.no_random:
# Where is the DLL? If missing, build using: `odin build . -build-mode:shared`
#
if platform.system() == "Windows":
- LIB_PATH = os.getcwd() + os.sep + "big.dll"
+ LIB_PATH = os.getcwd() + os.sep + "test_library.dll"
elif platform.system() == "Linux":
- LIB_PATH = os.getcwd() + os.sep + "big.so"
+ LIB_PATH = os.getcwd() + os.sep + "test_library.so"
elif platform.system() == "Darwin":
- LIB_PATH = os.getcwd() + os.sep + "big.dylib"
+ LIB_PATH = os.getcwd() + os.sep + "test_library.dylib"
else:
print("Platform is unsupported.")
exit(1)