nim-stint/tests/test_int_modular_arithmetic...

56 lines
2.3 KiB
Nim
Raw Normal View History

# Stint
# Copyright 2018-Present Status Research & Development GmbH
# Licensed under either of
#
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or http://www.apache.org/licenses/LICENSE-2.0)
# * MIT license ([LICENSE-MIT](LICENSE-MIT) or http://opensource.org/licenses/MIT)
#
# at your option. This file may not be copied, modified, or distributed except according to those terms.
import ../stint, unittest, math, test_helpers
template chkAddMod(chk: untyped, a, b, m, c: string, bits: int) =
chk addmod(fromHex(StInt[bits], a), fromHex(StInt[bits], b), fromHex(StInt[bits], m)) == fromHex(StInt[bits], c)
template chkMulMod(chk: untyped, a, b, m, c: string, bits: int) =
chk mulmod(fromHex(StInt[bits], a), fromHex(StInt[bits], b), fromHex(StInt[bits], m)) == fromHex(StInt[bits], c)
template chkPowMod(chk: untyped, a, b, m, c: string, bits: int) =
chk powmod(fromHex(StInt[bits], a), fromHex(StInt[bits], b), fromHex(StInt[bits], m)) == fromHex(StInt[bits], c)
template testModArith(chk, tst: untyped) =
tst "addmod":
chkAddMod(chk, "F", "F", "7", "2", 128)
chkAddMod(chk, "AAAA", "AA", "F", "0", 128)
chkAddMod(chk, "BBBB", "AAAA", "9", "3", 128)
chkAddMod(chk, "BBBBBBBB", "AAAAAAAA", "9", "6", 128)
chkAddMod(chk, "BBBBBBBBBBBBBBBB", "AAAAAAAAAAAAAAAA", "9", "3", 128)
chk addmod(-5.i128, -5.i128, 3.i128) == -1.i128
chk addmod(5.i128, -9.i128, 3.i128) == -1.i128
chk addmod(-5.i128, 9.i128, 3.i128) == 1.i128
tst "submod":
chk submod(10.i128, 5.i128, 3.i128) == 2.i128
chk submod(-6.i128, -5.i128, 3.i128) == -1.i128
chk submod(5.i128, -9.i128, 3.i128) == 2.i128
chk submod(-5.i128, 9.i128, 3.i128) == -2.i128
tst "mulmod":
chk mulmod(10.i128, 5.i128, 3.i128) == 2.i128
chk mulmod(-7.i128, -5.i128, 3.i128) == 2.i128
chk mulmod(6.i128, -9.i128, 4.i128) == -2.i128
chk mulmod(-5.i128, 7.i128, 3.i128) == -2.i128
tst "powmod":
chk powmod(10.i128, 5.i128, 3.i128) == 1.i128
chk powmod(-7.i128, 4.i128, 3.i128) == 1.i128
chk powmod(-7.i128, 3.i128, 3.i128) == -1.i128
chk powmod(5.i128, 9.i128, 3.i128) == 2.i128
chk powmod(-5.i128, 9.i128, 3.i128) == -2.i128
static:
testModArith(ctCheck, ctTest)
suite "Wider unsigned Modular arithmetic coverage":
testModArith(check, test)