int
main(void)
{
int i, result;
FLINT_TEST_INIT(state);
flint_printf("scalar_submul_si_2exp....");
fflush(stdout);
/* Compare with alternative method of computation */
for (i = 0; i < 1000 * flint_test_multiplier(); i++)
{
fmpz *a, *b, *c, *d;
slong len = n_randint(state, 100), x;
mp_bitcnt_t exp;
a = _fmpz_vec_init(len);
b = _fmpz_vec_init(len);
c = _fmpz_vec_init(len);
d = _fmpz_vec_init(len);
_fmpz_vec_randtest(a, state, len, 200);
_fmpz_vec_randtest(b, state, len, 200);
_fmpz_vec_set(c, b, len);
x = z_randtest(state);
exp = n_randint(state, 200);
_fmpz_vec_scalar_submul_si_2exp(b, a, len, x, exp);
_fmpz_vec_scalar_mul_2exp(d, a, len, exp);
_fmpz_vec_scalar_submul_si(c, d, len, x);
result = (_fmpz_vec_equal(b, c, len));
if (!result)
{
flint_printf("FAIL:\n");
flint_printf("x = %wd, exp = %wu\n", x, exp);
_fmpz_vec_print(b, len), flint_printf("\n\n");
_fmpz_vec_print(c, len), flint_printf("\n\n");
abort();
}
_fmpz_vec_clear(a, len);
_fmpz_vec_clear(b, len);
_fmpz_vec_clear(c, len);
_fmpz_vec_clear(d, len);
}
FLINT_TEST_CLEANUP(state);
flint_printf("PASS\n");
return 0;
}
/*
Returns 1 if sign * (G, glen) * (Q, qlen) = (P, len), else returns 0.
Temp requires space for glen + qlen - 1 coefficients
*/
int multiplies_out(fmpz * P, long len, const fmpz * Q, long qlen,
const fmpz * G, long glen, long sign, fmpz * temp)
{
int divides = 0;
/* multiply out */
if (qlen >= glen)
_fmpz_poly_mul(temp, Q, qlen, G, glen);
else
_fmpz_poly_mul(temp, G, glen, Q, qlen);
if (sign < 0L) _fmpz_vec_neg(temp, temp, glen + qlen - 1);
/* check if quotient really was exact */
divides = (glen + qlen - 1 == len && _fmpz_vec_equal(temp, P, len));
return divides;
}
int fmpz_mat_equal(const fmpz_mat_t mat1, const fmpz_mat_t mat2)
{
long j;
if (mat1->r != mat2->r || mat1->c != mat2->c)
{
return 0;
}
if (mat1->r == 0 || mat1->c == 0)
return 1;
for (j = 0; j < mat1->r; j++)
{
if (!_fmpz_vec_equal(mat1->rows[j], mat2->rows[j], mat1->c))
{
return 0;
}
}
return 1;
}
int
main(void)
{
int i, result;
flint_rand_t state;
printf("get/set_fft....");
fflush(stdout);
flint_randinit(state);
/* convert back and forth and compare */
for (i = 0; i < 10000; i++)
{
fmpz * a, * b;
mp_bitcnt_t bits;
long len, limbs;
mp_limb_t ** ii, * ptr;
long i, bt;
bits = n_randint(state, 300) + 1;
len = n_randint(state, 300) + 1;
limbs = 2*((bits - 1)/FLINT_BITS + 1);
ii = flint_malloc((len + len*(limbs + 1))*sizeof(mp_limb_t));
ptr = (mp_limb_t *) ii + len;
for (i = 0; i < len; i++, ptr += (limbs + 1))
ii[i] = ptr;
a = _fmpz_vec_init(len);
b = _fmpz_vec_init(len);
_fmpz_vec_randtest(a, state, len, bits);
bt = _fmpz_vec_get_fft(ii, a, limbs, len);
for (i = 0; i < len; i++)
mpn_normmod_2expp1(ii[i], limbs);
_fmpz_vec_set_fft(b, len, ii, limbs, bt < 0);
result = (_fmpz_vec_equal(a, b, len));
if (!result)
{
printf("FAIL:\n");
_fmpz_vec_print(a, len), printf("\n\n");
_fmpz_vec_print(b, len), printf("\n\n");
abort();
}
_fmpz_vec_clear(a, len);
_fmpz_vec_clear(b, len);
}
/* convert back and forth unsigned and compare */
for (i = 0; i < 10000; i++)
{
fmpz * a, * b;
mp_bitcnt_t bits;
long len, limbs;
mp_limb_t ** ii, * ptr;
long i, bt;
bits = n_randint(state, 300) + 1;
len = n_randint(state, 300) + 1;
limbs = 2*((bits - 1)/FLINT_BITS + 1);
ii = flint_malloc((len + len*(limbs + 1))*sizeof(mp_limb_t));
ptr = (mp_limb_t *) ii + len;
for (i = 0; i < len; i++, ptr += (limbs + 1))
ii[i] = ptr;
a = _fmpz_vec_init(len);
b = _fmpz_vec_init(len);
_fmpz_vec_randtest_unsigned(a, state, len, bits);
bt = _fmpz_vec_get_fft(ii, a, limbs, len);
_fmpz_vec_set_fft(b, len, ii, limbs, bt < 0);
result = (_fmpz_vec_equal(a, b, len));
if (!result)
{
printf("FAIL:\n");
_fmpz_vec_print(a, len), printf("\n\n");
_fmpz_vec_print(b, len), printf("\n\n");
abort();
}
_fmpz_vec_clear(a, len);
_fmpz_vec_clear(b, len);
}
flint_randclear(state);
_fmpz_cleanup();
printf("PASS\n");
return 0;
}
int
main(void)
{
int i, result;
FLINT_TEST_INIT(state);
flint_printf("scalar_divexact_ui....");
fflush(stdout);
/* Check aliasing of a and b */
for (i = 0; i < 1000 * flint_test_multiplier(); i++)
{
fmpz *a, *b;
ulong n = n_randtest_not_zero(state);
slong len = n_randint(state, 100);
a = _fmpz_vec_init(len);
b = _fmpz_vec_init(len);
_fmpz_vec_randtest(a, state, len, 200);
_fmpz_vec_scalar_mul_ui(a, a, len, n);
_fmpz_vec_scalar_divexact_ui(b, a, len, n);
_fmpz_vec_scalar_divexact_ui(a, a, len, n);
result = (_fmpz_vec_equal(a, b, len));
if (!result)
{
flint_printf("FAIL:\n");
_fmpz_vec_print(a, len), flint_printf("\n\n");
_fmpz_vec_print(b, len), flint_printf("\n\n");
abort();
}
_fmpz_vec_clear(a, len);
_fmpz_vec_clear(b, len);
}
/* Check that a * n / n == a */
for (i = 0; i < 1000 * flint_test_multiplier(); i++)
{
fmpz *a, *b;
ulong n = n_randtest_not_zero(state);
slong len = n_randint(state, 100);
a = _fmpz_vec_init(len);
b = _fmpz_vec_init(len);
_fmpz_vec_randtest(a, state, len, 200);
_fmpz_vec_set(b, a, len);
_fmpz_vec_scalar_mul_ui(a, a, len, n);
_fmpz_vec_scalar_divexact_ui(a, a, len, n);
result = (_fmpz_vec_equal(a, b, len));
if (!result)
{
flint_printf("FAIL:\n");
_fmpz_vec_print(a, len), flint_printf("\n\n");
_fmpz_vec_print(b, len), flint_printf("\n\n");
abort();
}
_fmpz_vec_clear(a, len);
_fmpz_vec_clear(b, len);
}
FLINT_TEST_CLEANUP(state);
flint_printf("PASS\n");
return 0;
}
int
main(void)
{
int i, result;
flint_rand_t state;
printf("scalar_submul_fmpz....");
fflush(stdout);
flint_randinit(state);
/* Compare with fmpz_vec_scalar_submul_si */
for (i = 0; i < 10000; i++)
{
fmpz *a, *b, *c;
long len, n;
fmpz_t n1;
len = n_randint(state, 100);
n = (long) n_randbits(state, FLINT_BITS - 1);
if (n_randint(state, 2))
n = -n;
fmpz_init(n1);
fmpz_set_si(n1, n);
a = _fmpz_vec_init(len);
b = _fmpz_vec_init(len);
c = _fmpz_vec_init(len);
_fmpz_vec_randtest(a, state, len, 200);
_fmpz_vec_randtest(b, state, len, 200);
_fmpz_vec_set(c, b, len);
_fmpz_vec_scalar_submul_fmpz(b, a, len, n1);
_fmpz_vec_scalar_submul_si(c, a, len, n);
result = (_fmpz_vec_equal(c, b, len));
if (!result)
{
printf("FAIL:\n");
_fmpz_vec_print(c, len), printf("\n\n");
_fmpz_vec_print(b, len), printf("\n\n");
abort();
}
fmpz_clear(n1);
_fmpz_vec_clear(a, len);
_fmpz_vec_clear(b, len);
_fmpz_vec_clear(c, len);
}
/* Compute a different way */
for (i = 0; i < 10000; i++)
{
fmpz *a, *b, *c, *d;
long len = n_randint(state, 100);
fmpz_t n1;
fmpz_init(n1);
fmpz_randtest(n1, state, 200);
a = _fmpz_vec_init(len);
b = _fmpz_vec_init(len);
c = _fmpz_vec_init(len);
d = _fmpz_vec_init(len);
_fmpz_vec_randtest(a, state, len, 200);
_fmpz_vec_randtest(b, state, len, 200);
_fmpz_vec_set(c, b, len);
_fmpz_vec_scalar_submul_fmpz(b, a, len, n1);
_fmpz_vec_scalar_mul_fmpz(d, a, len, n1);
_fmpz_vec_sub(c, c, d, len);
result = (_fmpz_vec_equal(c, b, len));
if (!result)
{
printf("FAIL:\n");
_fmpz_vec_print(c, len), printf("\n\n");
_fmpz_vec_print(b, len), printf("\n\n");
abort();
}
fmpz_clear(n1);
_fmpz_vec_clear(a, len);
_fmpz_vec_clear(b, len);
_fmpz_vec_clear(c, len);
_fmpz_vec_clear(d, len);
}
flint_randclear(state);
_fmpz_cleanup();
printf("PASS\n");
return 0;
}
int
main(void)
{
int i, result;
flint_rand_t state;
printf("scalar_mul_2exp....");
fflush(stdout);
flint_randinit(state);
/* Check aliasing of a and b */
for (i = 0; i < 10000; i++)
{
fmpz *a, *b;
long len = n_randint(state, 100);
ulong exp = n_randint(state, 200);
a = _fmpz_vec_init(len);
b = _fmpz_vec_init(len);
_fmpz_vec_randtest(a, state, len, 200);
_fmpz_vec_scalar_mul_2exp(b, a, len, exp);
_fmpz_vec_scalar_mul_2exp(a, a, len, exp);
result = (_fmpz_vec_equal(a, b, len));
if (!result)
{
printf("FAIL:\n");
printf("exp = %lu\n", exp);
_fmpz_vec_print(a, len), printf("\n\n");
_fmpz_vec_print(b, len), printf("\n\n");
abort();
}
_fmpz_vec_clear(a, len);
_fmpz_vec_clear(b, len);
}
/* Check aliasing of (a*2^e1)*2^e2 equals a*2^(e1+e2) */
for (i = 0; i < 10000; i++)
{
fmpz *a, *b;
long len = n_randint(state, 100);
ulong e1 = n_randint(state, 200);
ulong e2 = n_randint(state, 200);
a = _fmpz_vec_init(len);
b = _fmpz_vec_init(len);
_fmpz_vec_randtest(a, state, len, 200);
_fmpz_vec_scalar_mul_2exp(b, a, len, e1);
_fmpz_vec_scalar_mul_2exp(b, b, len, e2);
_fmpz_vec_scalar_mul_2exp(a, a, len, e1 + e2);
result = (_fmpz_vec_equal(a, b, len));
if (!result)
{
printf("FAIL:\n");
printf("e1 = %lu, e2 = %lu\n", e1, e2);
_fmpz_vec_print(a, len), printf("\n\n");
_fmpz_vec_print(b, len), printf("\n\n");
abort();
}
_fmpz_vec_clear(a, len);
_fmpz_vec_clear(b, len);
}
flint_randclear(state);
_fmpz_cleanup();
printf("PASS\n");
return 0;
}
