#include "test_helpers.h"
#include "fmpz.h"
#include "fmpz_mod.h"
TEST_FUNCTION_START(fmpz_mod_pow_fmpz, state)
{
flint_bitcnt_t max_exp_bits = 200;
flint_bitcnt_t max_modulus_bits = 200;
slong i, j;
for (i = 0; i < 20 * flint_test_multiplier(); i++)
{
fmpz_t p;
fmpz_t a, b, u, v, w, x;
fmpz_t e, f, t;
fmpz_mod_ctx_t fpctx;
fmpz_init_set_ui(p, 2);
fmpz_mod_ctx_init(fpctx, p);
fmpz_init(a);
fmpz_init(b);
fmpz_init(u);
fmpz_init(v);
fmpz_init(w);
fmpz_init(x);
fmpz_init(e);
fmpz_init(f);
fmpz_init(t);
for (j = 0; j < 100; j++)
{
if (n_randint(state, 1000) == 0)
fmpz_randtest_unsigned(p, state, 100000);
else
fmpz_randtest_unsigned(p, state, max_modulus_bits);
fmpz_add_ui(p, p, 1);
fmpz_mod_ctx_set_modulus(fpctx, p);
fmpz_randtest_mod(a, state, p);
fmpz_randtest_mod(b, state, p);
fmpz_randtest(e, state, max_exp_bits);
fmpz_randtest(f, state, max_exp_bits);
if (!fmpz_mod_pow_fmpz(u, a, e, fpctx))
goto skip1;
if (!fmpz_mod_pow_fmpz(v, a, f, fpctx))
goto skip1;
fmpz_add(t, e, f);
if (!fmpz_mod_pow_fmpz(w, a, t, fpctx))
goto skip1;
fmpz_mod_mul(x, u, v, fpctx);
if (!fmpz_equal(w, x))
{
printf("FAIL\n");
flint_printf("check a^(e + f) = a^e * a^f\n"
"i = %wd, j = %wd\n", i, j);
fflush(stdout);
flint_abort();
}
skip1:
if (!fmpz_mod_pow_fmpz(u, a, e, fpctx))
goto skip2;
if (!fmpz_mod_pow_fmpz(v, a, f, fpctx))
goto skip2;
fmpz_add(t, e, f);
if (!fmpz_mod_pow_fmpz(a, a, t, fpctx))
goto skip2;
fmpz_mod_mul(x, u, v, fpctx);
if (!fmpz_equal(a, x))
{
printf("FAIL\n");
flint_printf("check a^(e + f) = a^e * a^f with aliasing\n"
"i = %wd, j = %wd\n", i, j);
fflush(stdout);
flint_abort();
}
skip2:
fmpz_zero(a);
fmpz_zero(e);
fmpz_mod_pow_fmpz(u, a, e, fpctx);
if (!fmpz_mod_is_one(u, fpctx))
{
printf("FAIL\n");
flint_printf("check 0^0 = 1\n"
"i = %wd, j = %wd\n", i, j);
fflush(stdout);
flint_abort();
}
fmpz_mod_mul(u, a, b, fpctx);
if (!fmpz_mod_pow_fmpz(a, a, e, fpctx))
goto skip3;
if (!fmpz_mod_pow_fmpz(b, b, e, fpctx))
goto skip3;
if (!fmpz_mod_pow_fmpz(u, u, e, fpctx))
goto skip3;
fmpz_mod_mul(a, a, b, fpctx);
if (!fmpz_equal(a, u))
{
printf("FAIL\n");
flint_printf("check (a*b)^e = a^e * b^e with aliasing\n"
"i = %wd, j = %wd\n", i, j);
fflush(stdout);
flint_abort();
}
skip3:;
}
fmpz_clear(p);
fmpz_mod_ctx_clear(fpctx);
fmpz_clear(a);
fmpz_clear(b);
fmpz_clear(u);
fmpz_clear(v);
fmpz_clear(w);
fmpz_clear(x);
fmpz_clear(e);
fmpz_clear(f);
fmpz_clear(t);
}
TEST_FUNCTION_END(state);
}