# 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)