#include "test_helpers.h"
#include "mpn_extras.h"
TEST_FUNCTION_START(flint_mpn_divrem_preinv1, state)
{
int i, result;
mpz_t a, a2, b, q, r, q2;
gmp_randstate_t st;
mp_limb_t d1, d2, inv;
slong s1, s2;
mpz_init(a);
mpz_init(a2);
mpz_init(b);
mpz_init(q);
mpz_init(r);
gmp_randinit_default(st);
for (i = 0; i < 1000 * flint_test_multiplier(); i++)
{
do {
mpz_rrandomb(a, st, n_randint(state, 200));
do {
mpz_rrandomb(b, st, n_randint(state, 200));
} while (mpz_sgn(b) == 0);
s1 = a->_mp_size;
s2 = b->_mp_size;
} while (s1 < s2 || s2 < 2);
mpz_set(a2, a);
b->_mp_d[b->_mp_size - 1] |= ((mp_limb_t) 1 << (GMP_LIMB_BITS - 1));
d1 = b->_mp_d[b->_mp_size - 1];
d2 = b->_mp_d[b->_mp_size - 2];
mpz_fdiv_qr(q, r, a, b);
inv = flint_mpn_preinv1(d1, d2);
q2->_mp_d = flint_malloc((s1 - s2 + 1)*sizeof(mp_limb_t));
q2->_mp_d[s1 - s2] = flint_mpn_divrem_preinv1(q2->_mp_d, a2->_mp_d, a2->_mp_size, b->_mp_d, b->_mp_size, inv);
s1 -= (s2 - 1);
while (s1 && q2->_mp_d[s1 - 1] == 0) s1--;
q2->_mp_size = s1;
q2->_mp_alloc = s1;
while (s2 && a2->_mp_d[s2 - 1] == 0) s2--;
a2->_mp_size = s2;
result = (mpz_cmp(q, q2) == 0 && mpz_cmp(a2, r) == 0);
if (!result)
TEST_FUNCTION_FAIL(
"%{mpz}\n"
"%{mpz}\n"
"%{mpz}\n"
"%{mpz}\n"
"%{mpz}\n"
"%{mpz}\n",
a, b, q, r, q2, a2);
flint_free(q2->_mp_d);
}
mpz_clear(a);
mpz_clear(a2);
mpz_clear(b);
mpz_clear(q);
mpz_clear(r);
gmp_randclear(st);
TEST_FUNCTION_END(state);
}