package com.others;

import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;

import java.math.BigInteger;

class FastPowerTest {

    private void testLong(long n, long k, long m) {
        long result = FastPower.calculate(n, k, m);
        Assertions.assertEquals(result, BigInteger.valueOf(n).modPow(BigInteger.valueOf(k), BigInteger.valueOf(m)).longValue());
    }

    private void testBigInteger(BigInteger n, BigInteger k, BigInteger m) {
        BigInteger result = FastPower.calculate(n, k, m);
        Assertions.assertEquals(result, n.modPow(k, m));
    }

    @Test
    void test() {
        testLong(2, 2, 10);
        testLong(100, 1000, 20);
        testLong(123456, 123456789, 234);

        testBigInteger(BigInteger.TEN, BigInteger.TEN, BigInteger.valueOf(4));
        testBigInteger(new BigInteger("123456"), new BigInteger("123456789"), new BigInteger("234"));
        testBigInteger(new BigInteger("123456789101112"), new BigInteger("12345678910111213"), new BigInteger("567890"));

    }
}