int main()
{
slong iter;
flint_rand_t state;
flint_printf("exp_series_basecase....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 10000 * arb_test_multiplier(); iter++)
{
slong m, n, bits1, bits2, bits3;
acb_poly_t a, b, c, d;
bits1 = 2 + n_randint(state, 100);
bits2 = 2 + n_randint(state, 100);
bits3 = 2 + n_randint(state, 100);
m = 1 + n_randint(state, 30);
n = 1 + n_randint(state, 30);
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
acb_poly_init(d);
acb_poly_randtest(a, state, m, bits1, 5);
acb_poly_randtest(b, state, m, bits1, 5);
/* check exp(a+b) = exp(a) exp(b) */
acb_poly_exp_series_basecase(c, a, n, bits2);
acb_poly_exp_series_basecase(d, b, n, bits2);
acb_poly_mullow(c, c, d, n, bits2);
acb_poly_add(d, a, b, bits3);
acb_poly_exp_series_basecase(d, d, n, bits3); /* also aliasing test */
if (!acb_poly_overlaps(c, d))
{
flint_printf("FAIL\n\n");
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("c = "); acb_poly_printd(c, 15); flint_printf("\n\n");
flint_printf("d = "); acb_poly_printd(d, 15); flint_printf("\n\n");
abort();
}
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
acb_poly_clear(d);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("evaluate2....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 10000; iter++)
{
acb_poly_t f, g;
acb_t x, y1, z1, y2, z2;
acb_init(x);
acb_init(y1);
acb_init(z1);
acb_init(y2);
acb_init(z2);
acb_poly_init(f);
acb_poly_init(g);
acb_randtest(x, state, 2 + n_randint(state, 1000), 5);
acb_poly_randtest(f, state, 2 + n_randint(state, 100), 2 + n_randint(state, 1000), 5);
acb_poly_derivative(g, f, 2 + n_randint(state, 1000));
acb_poly_evaluate2(y1, z1, f, x, 2 + n_randint(state, 1000));
acb_poly_evaluate_horner(y2, f, x, 2 + n_randint(state, 1000));
acb_poly_evaluate_horner(z2, g, x, 2 + n_randint(state, 1000));
if (!acb_overlaps(y1, y2) || !acb_overlaps(z1, z2))
{
flint_printf("FAIL\n\n");
flint_printf("f = "); acb_poly_printd(f, 15); flint_printf("\n\n");
flint_printf("g = "); acb_poly_printd(g, 15); flint_printf("\n\n");
flint_printf("x = "); acb_printd(x, 15); flint_printf("\n\n");
flint_printf("y1 = "); acb_printd(y1, 15); flint_printf("\n\n");
flint_printf("z1 = "); acb_printd(z1, 15); flint_printf("\n\n");
flint_printf("y2 = "); acb_printd(y2, 15); flint_printf("\n\n");
flint_printf("z2 = "); acb_printd(z2, 15); flint_printf("\n\n");
abort();
}
acb_poly_clear(f);
acb_poly_clear(g);
acb_clear(x);
acb_clear(y1);
acb_clear(z1);
acb_clear(y2);
acb_clear(z2);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("evaluate_rectangular....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 10000 * arb_test_multiplier(); iter++)
{
acb_poly_t f;
acb_t x, y1, y2;
acb_init(x);
acb_init(y1);
acb_init(y2);
acb_poly_init(f);
acb_randtest(x, state, 2 + n_randint(state, 1000), 5);
acb_poly_randtest(f, state, 2 + n_randint(state, 100), 2 + n_randint(state, 1000), 5);
acb_poly_evaluate_rectangular(y1, f, x, 2 + n_randint(state, 1000));
acb_poly_evaluate_horner(y2, f, x, 2 + n_randint(state, 1000));
if (!acb_overlaps(y1, y2))
{
flint_printf("FAIL\n\n");
flint_printf("f = "); acb_poly_printd(f, 15); flint_printf("\n\n");
flint_printf("x = "); acb_printd(x, 15); flint_printf("\n\n");
flint_printf("y1 = "); acb_printd(y1, 15); flint_printf("\n\n");
flint_printf("y2 = "); acb_printd(y2, 15); flint_printf("\n\n");
abort();
}
acb_poly_clear(f);
acb_clear(x);
acb_clear(y1);
acb_clear(y2);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int
main(void)
{
int i, result;
flint_rand_t state;
printf("get_coeff_ptr....");
fflush(stdout);
flint_randinit(state);
for (i = 0; i < 1000; i++)
{
acb_poly_t A;
acb_t a;
long n = n_randint(state, 100);
acb_poly_init(A);
acb_poly_randtest(A, state, n_randint(state, 100), 100, 10);
acb_init(a);
acb_poly_get_coeff_acb(a, A, n);
result = n < acb_poly_length(A) ?
acb_equal(a, acb_poly_get_coeff_ptr(A, n)) :
acb_poly_get_coeff_ptr(A, n) == NULL;
if (!result)
{
printf("FAIL:\n");
printf("A = "), acb_poly_printd(A, 10), printf("\n\n");
printf("a = "), acb_print(a), printf("\n\n");
printf("n = %ld\n\n", n);
abort();
}
acb_poly_clear(A);
acb_clear(a);
}
flint_randclear(state);
flint_cleanup();
printf("PASS\n");
return 0;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("pow_acb_series....");
fflush(stdout);
flint_randinit(state);
/* compare with exp/log */
for (iter = 0; iter < 10000 * arb_test_multiplier(); iter++)
{
slong prec, trunc;
acb_poly_t f, g, h1, h2;
acb_t c;
prec = 2 + n_randint(state, 200);
trunc = n_randint(state, 20);
acb_poly_init(f);
acb_poly_init(g);
acb_poly_init(h1);
acb_poly_init(h2);
acb_init(c);
/* generate binomials */
if (n_randint(state, 20) == 0)
{
acb_randtest(c, state, prec, 10);
acb_poly_set_coeff_acb(f, 0, c);
acb_randtest(c, state, prec, 10);
acb_poly_set_coeff_acb(f, 1 + n_randint(state, 20), c);
}
else
{
acb_poly_randtest(f, state, 1 + n_randint(state, 20), prec, 10);
}
acb_poly_randtest(h1, state, 1 + n_randint(state, 20), prec, 10);
acb_randtest(c, state, prec, 10);
acb_poly_set_acb(g, c);
/* f^c */
acb_poly_pow_acb_series(h1, f, c, trunc, prec);
/* f^c = exp(c*log(f)) */
acb_poly_log_series(h2, f, trunc, prec);
acb_poly_mullow(h2, h2, g, trunc, prec);
acb_poly_exp_series(h2, h2, trunc, prec);
if (!acb_poly_overlaps(h1, h2))
{
flint_printf("FAIL\n\n");
flint_printf("prec = %wd\n", prec);
flint_printf("trunc = %wd\n", trunc);
flint_printf("f = "); acb_poly_printd(f, 15); flint_printf("\n\n");
flint_printf("c = "); acb_printd(c, 15); flint_printf("\n\n");
flint_printf("h1 = "); acb_poly_printd(h1, 15); flint_printf("\n\n");
flint_printf("h2 = "); acb_poly_printd(h2, 15); flint_printf("\n\n");
abort();
}
acb_poly_pow_acb_series(f, f, c, trunc, prec);
if (!acb_poly_overlaps(f, h1))
{
flint_printf("FAIL (aliasing)\n\n");
abort();
}
acb_poly_clear(f);
acb_poly_clear(g);
acb_poly_clear(h1);
acb_poly_clear(h2);
acb_clear(c);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("add_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 1000 * arb_test_multiplier(); iter++)
{
acb_poly_t a, b, c, d;
slong len, prec;
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
acb_poly_init(d);
acb_poly_randtest(a, state, 1 + n_randint(state, 10), 100, 10);
acb_poly_randtest(b, state, 1 + n_randint(state, 10), 100, 10);
acb_poly_randtest(c, state, 1 + n_randint(state, 10), 100, 10);
acb_poly_randtest(d, state, 1 + n_randint(state, 10), 100, 10);
prec = 2 + n_randint(state, 100);
len = n_randint(state, 10);
acb_poly_add_series(c, a, b, len, prec);
acb_poly_add(d, a, b, prec);
acb_poly_truncate(d, len);
if (!acb_poly_equal(c, d))
{
flint_printf("FAIL\n\n");
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("c = "); acb_poly_printd(c, 15); flint_printf("\n\n");
flint_printf("c = "); acb_poly_printd(c, 15); flint_printf("\n\n");
flint_abort();
}
acb_poly_set(d, a);
acb_poly_add_series(d, d, b, len, prec);
if (!acb_poly_equal(d, c))
{
flint_printf("FAIL (aliasing 1)\n\n");
flint_abort();
}
acb_poly_set(d, b);
acb_poly_add_series(d, a, d, len, prec);
if (!acb_poly_equal(d, c))
{
flint_printf("FAIL (aliasing 2)\n\n");
flint_abort();
}
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
acb_poly_clear(d);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("compose....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 3000 * arb_test_multiplier(); iter++)
{
slong qbits1, qbits2, rbits1, rbits2, rbits3;
fmpq_poly_t A, B, C;
acb_poly_t a, b, c, d;
qbits1 = 2 + n_randint(state, 200);
qbits2 = 2 + n_randint(state, 200);
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
rbits3 = 2 + n_randint(state, 200);
fmpq_poly_init(A);
fmpq_poly_init(B);
fmpq_poly_init(C);
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
acb_poly_init(d);
fmpq_poly_randtest(A, state, 1 + n_randint(state, 20), qbits1);
fmpq_poly_randtest(B, state, 1 + n_randint(state, 10), qbits2);
fmpq_poly_compose(C, A, B);
acb_poly_randtest(c, state, 1 + n_randint(state, 20), rbits1, 4);
acb_poly_set_fmpq_poly(a, A, rbits1);
acb_poly_set_fmpq_poly(b, B, rbits2);
acb_poly_compose(c, a, b, rbits3);
if (!acb_poly_contains_fmpq_poly(c, C))
{
flint_printf("FAIL\n\n");
flint_printf("bits3 = %wd\n", rbits3);
flint_printf("A = "); fmpq_poly_print(A); flint_printf("\n\n");
flint_printf("B = "); fmpq_poly_print(B); flint_printf("\n\n");
flint_printf("C = "); fmpq_poly_print(C); flint_printf("\n\n");
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("c = "); acb_poly_printd(c, 15); flint_printf("\n\n");
flint_abort();
}
acb_poly_set(d, a);
acb_poly_compose(d, d, b, rbits3);
if (!acb_poly_equal(d, c))
{
flint_printf("FAIL (aliasing 1)\n\n");
flint_abort();
}
acb_poly_set(d, b);
acb_poly_compose(d, a, d, rbits3);
if (!acb_poly_equal(d, c))
{
flint_printf("FAIL (aliasing 2)\n\n");
flint_abort();
}
fmpq_poly_clear(A);
fmpq_poly_clear(B);
fmpq_poly_clear(C);
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
acb_poly_clear(d);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
long iter;
flint_rand_t state;
printf("log_series....");
fflush(stdout);
flint_randinit(state);
/* compare with fmpq_poly */
for (iter = 0; iter < 10000; iter++)
{
long m, n, qbits, rbits1, rbits2;
fmpq_poly_t A, B;
acb_poly_t a, b;
qbits = 2 + n_randint(state, 200);
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 20);
n = 1 + n_randint(state, 20);
fmpq_poly_init(A);
fmpq_poly_init(B);
acb_poly_init(a);
acb_poly_init(b);
fmpq_poly_randtest_not_zero(A, state, m, qbits);
fmpq_poly_set_coeff_ui(A, 0, 1UL);
fmpq_poly_log_series(B, A, n);
acb_poly_set_fmpq_poly(a, A, rbits1);
acb_poly_randtest(b, state, 1 + n_randint(state, 20), rbits1, 5);
acb_poly_log_series(b, a, n, rbits2);
if (!acb_poly_contains_fmpq_poly(b, B))
{
printf("FAIL\n\n");
printf("bits2 = %ld\n", rbits2);
printf("A = "); fmpq_poly_print(A); printf("\n\n");
printf("B = "); fmpq_poly_print(B); printf("\n\n");
printf("a = "); acb_poly_printd(a, 15); printf("\n\n");
printf("b = "); acb_poly_printd(b, 15); printf("\n\n");
abort();
}
fmpq_poly_clear(A);
fmpq_poly_clear(B);
acb_poly_clear(a);
acb_poly_clear(b);
}
/* test aliasing */
for (iter = 0; iter < 10000; iter++)
{
long m, n, qbits, rbits1, rbits2;
fmpq_poly_t A;
acb_poly_t a, b;
qbits = 2 + n_randint(state, 200);
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 20);
n = 1 + n_randint(state, 20);
fmpq_poly_init(A);
acb_poly_init(a);
acb_poly_init(b);
do {
fmpq_poly_randtest_not_zero(A, state, m, qbits);
} while (fmpz_sgn(A->coeffs + 0) <= 0);
acb_poly_set_fmpq_poly(a, A, rbits1);
acb_poly_log_series(b, a, n, rbits2);
acb_poly_log_series(a, a, n, rbits2);
if (!acb_poly_equal(a, b))
{
printf("FAIL\n\n");
printf("bits2 = %ld\n", rbits2);
printf("A = "); fmpq_poly_print(A); printf("\n\n");
printf("a = "); acb_poly_printd(a, 15); printf("\n\n");
printf("b = "); acb_poly_printd(b, 15); printf("\n\n");
abort();
}
fmpq_poly_clear(A);
acb_poly_clear(a);
acb_poly_clear(b);
}
/* test that exp(log(f)) contains f */
for (iter = 0; iter < 10000; iter++)
{
long m, n, qbits, rbits1, rbits2, rbits3;
fmpq_poly_t A;
acb_poly_t a, b, c;
qbits = 2 + n_randint(state, 200);
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
rbits3 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 20);
n = 1 + n_randint(state, 20);
fmpq_poly_init(A);
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
do {
fmpq_poly_randtest_not_zero(A, state, m, qbits);
} while (fmpz_sgn(A->coeffs + 0) <= 0);
acb_poly_set_fmpq_poly(a, A, rbits1);
acb_poly_randtest(b, state, 1 + n_randint(state, 20), rbits1, 5);
acb_poly_log_series(b, a, n, rbits2);
acb_poly_exp_series_basecase(c, b, n, rbits3);
fmpq_poly_truncate(A, n);
if (!acb_poly_contains_fmpq_poly(c, A))
{
printf("FAIL\n\n");
printf("bits2 = %ld\n", rbits2);
printf("bits3 = %ld\n", rbits3);
printf("A = "); fmpq_poly_print(A); printf("\n\n");
printf("a = "); acb_poly_printd(a, 15); printf("\n\n");
printf("b = "); acb_poly_printd(b, 15); printf("\n\n");
printf("c = "); acb_poly_printd(c, 15); printf("\n\n");
abort();
}
fmpq_poly_clear(A);
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
}
flint_randclear(state);
flint_cleanup();
printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("revert_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 1000 * arb_test_multiplier(); iter++)
{
slong qbits1, rbits1, rbits2, n;
fmpq_poly_t A, B;
acb_poly_t a, b, c;
qbits1 = 2 + n_randint(state, 200);
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
n = 2 + n_randint(state, 25);
fmpq_poly_init(A);
fmpq_poly_init(B);
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
do {
fmpq_poly_randtest(A, state, 1 + n_randint(state, 25), qbits1);
fmpq_poly_set_coeff_ui(A, 0, 0);
} while (A->length < 2 || fmpz_is_zero(A->coeffs + 1));
fmpq_poly_revert_series(B, A, n);
acb_poly_set_fmpq_poly(a, A, rbits1);
acb_poly_revert_series(b, a, n, rbits2);
if (!acb_poly_contains_fmpq_poly(b, B))
{
flint_printf("FAIL\n\n");
flint_printf("n = %wd, bits2 = %wd\n", n, rbits2);
flint_printf("A = "); fmpq_poly_print(A); flint_printf("\n\n");
flint_printf("B = "); fmpq_poly_print(B); flint_printf("\n\n");
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
abort();
}
acb_poly_set(c, a);
acb_poly_revert_series(c, c, n, rbits2);
if (!acb_poly_equal(c, b))
{
flint_printf("FAIL (aliasing)\n\n");
abort();
}
fmpq_poly_clear(A);
fmpq_poly_clear(B);
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("taylor_shift_convolution....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 2000; iter++)
{
slong prec1, prec2;
acb_poly_t f, g;
acb_t c, d, e;
prec1 = 2 + n_randint(state, 500);
prec2 = 2 + n_randint(state, 500);
acb_poly_init(f);
acb_poly_init(g);
acb_init(c);
acb_init(d);
acb_init(e);
acb_poly_randtest(f, state, 1 + n_randint(state, 40), 1 + n_randint(state, 500), 10);
acb_poly_randtest(g, state, 1 + n_randint(state, 20), 1 + n_randint(state, 500), 10);
if (n_randint(state, 2))
acb_set_si(c, n_randint(state, 5) - 2);
else
acb_randtest(c, state, 1 + n_randint(state, 500), 1 + n_randint(state, 100));
if (n_randint(state, 2))
acb_set_si(d, n_randint(state, 5) - 2);
else
acb_randtest(d, state, 1 + n_randint(state, 500), 1 + n_randint(state, 100));
acb_add(e, c, d, prec1);
/* check f(x+c)(x+d) = f(x+c+d) */
acb_poly_taylor_shift_convolution(g, f, e, prec2);
acb_poly_taylor_shift_convolution(f, f, c, prec1);
acb_poly_taylor_shift_convolution(f, f, d, prec1);
if (!acb_poly_overlaps(f, g))
{
flint_printf("FAIL\n\n");
flint_printf("c = "); acb_printd(c, 15); flint_printf("\n\n");
flint_printf("d = "); acb_printd(d, 15); flint_printf("\n\n");
flint_printf("f = "); acb_poly_printd(f, 15); flint_printf("\n\n");
flint_printf("g = "); acb_poly_printd(g, 15); flint_printf("\n\n");
abort();
}
acb_poly_clear(f);
acb_poly_clear(g);
acb_clear(c);
acb_clear(d);
acb_clear(e);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("hardy_z_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 200 * arb_test_multiplier(); iter++)
{
slong m, n1, n2, bits1, bits2, bits3;
acb_poly_t S, A, B, C;
dirichlet_group_t G;
dirichlet_char_t chi;
ulong q;
bits1 = 2 + n_randint(state, 200);
bits2 = 2 + n_randint(state, 200);
bits3 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 8);
n1 = 1 + n_randint(state, 8);
n2 = 1 + n_randint(state, 8);
do {
q = 1 + n_randint(state, 15);
} while (q % 4 == 2);
dirichlet_group_init(G, q);
dirichlet_char_init(chi, G);
do {
dirichlet_char_index(chi, G, n_randint(state, G->phi_q));
} while (!dirichlet_char_is_primitive(G, chi));
acb_poly_init(S);
acb_poly_init(A);
acb_poly_init(B);
acb_poly_init(C);
acb_poly_randtest(S, state, m, bits1, 3);
acb_dirichlet_hardy_z_series(A, S, G, chi, n1, bits2);
acb_poly_set(B, S); /* aliasing */
if (q == 1 && n_randint(state, 2))
acb_poly_neg(B, B);
acb_dirichlet_hardy_z_series(B, B, G, chi, n2, bits3);
acb_poly_set(C, A);
acb_poly_truncate(C, FLINT_MIN(n1, n2));
acb_poly_truncate(B, FLINT_MIN(n1, n2));
if (!acb_poly_overlaps(B, C))
{
flint_printf("FAIL\n\n");
flint_printf("S = "); acb_poly_printd(S, 15); flint_printf("\n\n");
flint_printf("A = "); acb_poly_printd(A, 15); flint_printf("\n\n");
flint_printf("B = "); acb_poly_printd(B, 15); flint_printf("\n\n");
abort();
}
acb_poly_clear(S);
acb_poly_clear(A);
acb_poly_clear(B);
acb_poly_clear(C);
dirichlet_char_clear(chi);
dirichlet_group_clear(G);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("cot_pi_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 1000; iter++)
{
slong m, n1, bits1, bits2, bits3;
acb_poly_t S, A, B, C;
bits1 = 2 + n_randint(state, 200);
bits2 = 2 + n_randint(state, 200);
bits3 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 30);
n1 = 1 + n_randint(state, 30);
acb_poly_init(S);
acb_poly_init(A);
acb_poly_init(B);
acb_poly_init(C);
acb_poly_randtest(S, state, m, bits1, 10);
acb_poly_randtest(A, state, m, bits1, 10);
acb_poly_randtest(B, state, m, bits1, 10);
acb_poly_cot_pi_series(A, S, n1, bits2);
acb_poly_sin_cos_pi_series(B, C, S, n1, bits3);
acb_poly_div_series(B, C, B, n1, bits3);
if (!acb_poly_overlaps(A, B))
{
flint_printf("FAIL\n\n");
flint_printf("S = "); acb_poly_printd(S, 15); flint_printf("\n\n");
flint_printf("A = "); acb_poly_printd(A, 15); flint_printf("\n\n");
flint_printf("B = "); acb_poly_printd(B, 15); flint_printf("\n\n");
abort();
}
acb_poly_cot_pi_series(S, S, n1, bits2);
if (!acb_poly_overlaps(A, S))
{
flint_printf("FAIL (aliasing)\n\n");
abort();
}
acb_poly_clear(S);
acb_poly_clear(A);
acb_poly_clear(B);
acb_poly_clear(C);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("div_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 10000 * arb_test_multiplier(); iter++)
{
slong m, n, p, qbits, rbits1, rbits2;
fmpq_poly_t A, B, C;
acb_poly_t a, b, c, d;
qbits = 2 + n_randint(state, 200);
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 20);
n = 1 + n_randint(state, 20);
p = 1 + n_randint(state, 20);
fmpq_poly_init(A);
fmpq_poly_init(B);
fmpq_poly_init(C);
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
acb_poly_init(d);
fmpq_poly_randtest(A, state, m, qbits);
do {
fmpq_poly_randtest_not_zero(B, state, n, qbits);
} while (B->coeffs[0] == 0);
fmpq_poly_div_series(C, A, B, p);
acb_poly_set_fmpq_poly(a, A, rbits1);
acb_poly_set_fmpq_poly(b, B, rbits1);
acb_poly_div_series(c, a, b, p, rbits2);
if (!acb_poly_contains_fmpq_poly(c, C))
{
flint_printf("FAIL\n\n");
flint_printf("bits2 = %wd\n", rbits2);
flint_printf("A = ");
fmpq_poly_print(A);
flint_printf("\n\n");
flint_printf("B = ");
fmpq_poly_print(B);
flint_printf("\n\n");
flint_printf("C = ");
fmpq_poly_print(C);
flint_printf("\n\n");
flint_printf("a = ");
acb_poly_printd(a, 15);
flint_printf("\n\n");
flint_printf("b = ");
acb_poly_printd(b, 15);
flint_printf("\n\n");
flint_printf("c = ");
acb_poly_printd(c, 15);
flint_printf("\n\n");
abort();
}
acb_poly_set(d, a);
acb_poly_div_series(d, d, b, p, rbits2);
if (!acb_poly_equal(d, c))
{
flint_printf("FAIL (aliasing 1)\n\n");
abort();
}
acb_poly_set(d, b);
acb_poly_div_series(d, a, d, p, rbits2);
if (!acb_poly_equal(d, c))
{
flint_printf("FAIL (aliasing 2)\n\n");
abort();
}
acb_poly_set(d, b);
acb_poly_div_series(c, d, d, p, rbits2);
acb_poly_div_series(d, d, d, p, rbits2);
if (!acb_poly_equal(d, c))
{
flint_printf("FAIL (aliasing 3)\n\n");
abort();
}
fmpq_poly_clear(A);
fmpq_poly_clear(B);
fmpq_poly_clear(C);
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
acb_poly_clear(d);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
long iter;
flint_rand_t state;
printf("compose_series_brent_kung....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 3000; iter++)
{
long qbits1, qbits2, rbits1, rbits2, rbits3, n;
fmpq_poly_t A, B, C;
acb_poly_t a, b, c, d;
qbits1 = 2 + n_randint(state, 200);
qbits2 = 2 + n_randint(state, 200);
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
rbits3 = 2 + n_randint(state, 200);
n = 2 + n_randint(state, 25);
fmpq_poly_init(A);
fmpq_poly_init(B);
fmpq_poly_init(C);
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
acb_poly_init(d);
fmpq_poly_randtest(A, state, 1 + n_randint(state, 25), qbits1);
fmpq_poly_randtest(B, state, 1 + n_randint(state, 25), qbits2);
fmpq_poly_set_coeff_ui(B, 0, 0);
fmpq_poly_compose_series(C, A, B, n);
acb_poly_set_fmpq_poly(a, A, rbits1);
acb_poly_set_fmpq_poly(b, B, rbits2);
acb_poly_compose_series_brent_kung(c, a, b, n, rbits3);
if (!acb_poly_contains_fmpq_poly(c, C))
{
printf("FAIL\n\n");
printf("n = %ld, bits3 = %ld\n", n, rbits3);
printf("A = "); fmpq_poly_print(A); printf("\n\n");
printf("B = "); fmpq_poly_print(B); printf("\n\n");
printf("C = "); fmpq_poly_print(C); printf("\n\n");
printf("a = "); acb_poly_printd(a, 15); printf("\n\n");
printf("b = "); acb_poly_printd(b, 15); printf("\n\n");
printf("c = "); acb_poly_printd(c, 15); printf("\n\n");
abort();
}
acb_poly_set(d, a);
acb_poly_compose_series_brent_kung(d, d, b, n, rbits3);
if (!acb_poly_equal(d, c))
{
printf("FAIL (aliasing 1)\n\n");
abort();
}
acb_poly_set(d, b);
acb_poly_compose_series_brent_kung(d, a, d, n, rbits3);
if (!acb_poly_equal(d, c))
{
printf("FAIL (aliasing 2)\n\n");
abort();
}
fmpq_poly_clear(A);
fmpq_poly_clear(B);
fmpq_poly_clear(C);
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
acb_poly_clear(d);
}
flint_randclear(state);
flint_cleanup();
printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("sinh_cosh_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 1000 * arb_test_multiplier(); iter++)
{
acb_poly_t a, b, c, d, e;
slong n1, n2, prec;
int alg1, alg2;
prec = 2 + n_randint(state, 200);
if (n_randint(state, 40) == 0)
{
n1 = n_randint(state, 300);
n2 = n_randint(state, 300);
}
else
{
n1 = n_randint(state, 30);
n2 = n_randint(state, 30);
}
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
acb_poly_init(d);
acb_poly_init(e);
acb_poly_randtest(a, state, 20, prec, 3);
acb_poly_randtest(b, state, 20, prec, 10);
acb_poly_randtest(c, state, 20, prec, 10);
acb_poly_randtest(d, state, 20, prec, 10);
acb_poly_randtest(e, state, 20, prec, 10);
alg1 = n_randint(state, 4);
alg2 = n_randint(state, 7);
switch (alg1)
{
case 0:
acb_poly_sinh_cosh_series(b, c, a, n1, prec);
break;
case 1:
acb_poly_sinh_cosh_series_basecase(b, c, a, n1, prec);
break;
case 2:
acb_poly_sinh_cosh_series_exponential(b, c, a, n1, prec);
break;
default:
acb_poly_sinh_series(b, a, n1, prec);
acb_poly_cosh_series(c, a, n1, prec);
break;
}
switch (alg2)
{
case 0:
acb_poly_sinh_cosh_series(d, e, a, n2, prec);
break;
case 1:
acb_poly_sinh_cosh_series_basecase(d, e, a, n2, prec);
break;
case 2:
acb_poly_sinh_cosh_series_exponential(d, e, a, n2, prec);
break;
case 3:
acb_poly_sinh_series(d, a, n2, prec);
acb_poly_cosh_series(e, a, n2, prec);
break;
case 4:
acb_poly_set(d, a);
acb_poly_sinh_cosh_series(d, e, d, n2, prec);
break;
case 5:
acb_poly_set(e, a);
acb_poly_sinh_cosh_series(d, e, e, n2, prec);
break;
default:
acb_poly_set(d, a);
acb_poly_sinh_series(d, d, n2, prec);
acb_poly_set(e, a);
acb_poly_cosh_series(e, e, n2, prec);
break;
}
acb_poly_truncate(b, FLINT_MIN(n1, n2));
acb_poly_truncate(c, FLINT_MIN(n1, n2));
acb_poly_truncate(d, FLINT_MIN(n1, n2));
acb_poly_truncate(e, FLINT_MIN(n1, n2));
if (!acb_poly_overlaps(b, d) || !acb_poly_overlaps(c, e))
{
flint_printf("FAIL\n\n");
flint_printf("alg1 = %d, alg2 = %d, n1 = %wd, n2 = %wd\n\n", alg1, alg2, n1, n2);
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("c = "); acb_poly_printd(c, 15); flint_printf("\n\n");
flint_printf("d = "); acb_poly_printd(d, 15); flint_printf("\n\n");
flint_printf("d = "); acb_poly_printd(e, 15); flint_printf("\n\n");
abort();
}
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
acb_poly_clear(d);
acb_poly_clear(e);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
long iter;
flint_rand_t state;
printf("polylog_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 100; iter++)
{
long m, n1, n2, bits1, bits2, bits3;
acb_poly_t S, A, B, C;
acb_t z;
bits1 = 2 + n_randint(state, 200);
bits2 = 2 + n_randint(state, 200);
bits3 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 10);
n1 = 1 + n_randint(state, 10);
n2 = 1 + n_randint(state, 10);
acb_poly_init(S);
acb_poly_init(A);
acb_poly_init(B);
acb_poly_init(C);
acb_init(z);
acb_poly_randtest(S, state, m, bits1, 3);
acb_poly_randtest(A, state, m, bits1, 3);
acb_poly_randtest(B, state, m, bits1, 3);
acb_randtest(z, state, bits1, 3);
acb_poly_polylog_series(A, S, z, n1, bits2);
acb_poly_polylog_series(B, S, z, n2, bits3);
acb_poly_set(C, A);
acb_poly_truncate(C, FLINT_MIN(n1, n2));
acb_poly_truncate(B, FLINT_MIN(n1, n2));
if (!acb_poly_overlaps(B, C))
{
printf("FAIL\n\n");
printf("S = "); acb_poly_printd(S, 15); printf("\n\n");
printf("z = "); acb_printd(z, 15); printf("\n\n");
printf("A = "); acb_poly_printd(A, 15); printf("\n\n");
printf("B = "); acb_poly_printd(B, 15); printf("\n\n");
abort();
}
acb_poly_polylog_series(S, S, z, n1, bits2);
if (!acb_poly_overlaps(A, S))
{
printf("FAIL (aliasing)\n\n");
abort();
}
acb_poly_clear(S);
acb_poly_clear(A);
acb_poly_clear(B);
acb_poly_clear(C);
acb_clear(z);
}
flint_randclear(state);
flint_cleanup();
printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("gamma_upper_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 1000 * arb_test_multiplier(); iter++)
{
slong m, n1, n2, bits1, bits2, bits3;
acb_poly_t S, A, B, C;
acb_t s, t, c;
int regularized;
regularized = n_randint(state, 3);
bits1 = 2 + n_randint(state, 200);
bits2 = 2 + n_randint(state, 200);
bits3 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 10);
n1 = 1 + n_randint(state, 10);
n2 = 1 + n_randint(state, 10);
acb_poly_init(S);
acb_poly_init(A);
acb_poly_init(B);
acb_poly_init(C);
acb_init(s);
acb_init(t);
acb_init(c);
acb_poly_randtest(S, state, m, bits1, 3);
acb_poly_randtest(A, state, m, bits1, 3);
acb_poly_randtest(B, state, m, bits1, 3);
acb_randtest(s, state, bits1, 3);
acb_hypgeom_gamma_upper_series(A, s, S, regularized, n1, bits2);
acb_hypgeom_gamma_upper_series(B, s, S, regularized, n2, bits3);
acb_poly_set(C, A);
acb_poly_truncate(C, FLINT_MIN(n1, n2));
acb_poly_truncate(B, FLINT_MIN(n1, n2));
if (!acb_poly_overlaps(B, C))
{
flint_printf("FAIL (consistency)\n\n");
flint_printf("regularized = %d\n\n", regularized);
flint_printf("S = "); acb_poly_printd(S, 15); flint_printf("\n\n");
flint_printf("A = "); acb_poly_printd(A, 15); flint_printf("\n\n");
flint_printf("B = "); acb_poly_printd(B, 15); flint_printf("\n\n");
flint_printf("C = "); acb_poly_printd(C, 15); flint_printf("\n\n");
flint_abort();
}
/* f(h(x)) = -exp(-h(x)) h(x)^(s-1) */
acb_poly_neg(C, S);
acb_poly_exp_series(C, C, n1, bits2);
acb_sub_ui(t, s, 1, bits2);
acb_poly_pow_acb_series(B, S, t, n1, bits2);
acb_poly_mullow(C, C, B, n1, bits2);
acb_poly_neg(C, C);
if (regularized == 0)
{
/* integral(f(h(x)) h'(x))' = f(h(x)) h'(x) */
acb_poly_derivative(B, S, bits2);
acb_poly_mullow(C, C, B, n1, bits2);
acb_poly_truncate(C, n1 - 1);
}
else if (regularized == 1)
{
/* (integral(f(h(x)) h'(x)) / c)' = (f(h(x)) h'(x)) / c */
acb_poly_derivative(B, S, bits2);
acb_poly_mullow(C, C, B, n1, bits2);
acb_gamma(c, s, bits2);
acb_poly_scalar_div(C, C, c, bits2);
acb_poly_truncate(C, n1 - 1);
}
else if (regularized == 2)
{
/* (h(x)^-s integral(f(h(x)) h'(x)))' =
* h(x)^-(s+1) (h(x) f(h(x)) - s integral(f(h(x)) h'(x))) h'(x) */
acb_poly_t D;
acb_poly_init(D);
acb_poly_derivative(B, S, bits2);
acb_poly_mullow(D, C, B, n1, bits2);
acb_poly_integral(D, D, bits2);
acb_poly_scalar_mul(D, D, s, bits2);
acb_poly_mullow(C, C, S, n1, bits2);
acb_poly_sub(D, C, D, bits2);
acb_add_ui(t, s, 1, bits2);
acb_neg(t, t);
acb_poly_pow_acb_series(B, S, t, n1, bits2);
acb_poly_mullow(C, D, B, n1, bits2);
acb_poly_derivative(B, S, bits2);
acb_poly_mullow(C, C, B, n1, bits2);
acb_poly_truncate(C, n1 - 1);
acb_poly_clear(D);
}
acb_poly_derivative(B, A, bits2);
if (!acb_poly_overlaps(B, C))
{
flint_printf("FAIL (derivative)\n\n");
flint_printf("regularized = %d\n\n", regularized);
flint_printf("S = "); acb_poly_printd(S, 15); flint_printf("\n\n");
flint_printf("A = "); acb_poly_printd(A, 15); flint_printf("\n\n");
flint_printf("B = "); acb_poly_printd(B, 15); flint_printf("\n\n");
flint_printf("C = "); acb_poly_printd(C, 15); flint_printf("\n\n");
flint_abort();
}
acb_hypgeom_gamma_upper_series(S, s, S, regularized, n1, bits2);
if (!acb_poly_overlaps(A, S))
{
flint_printf("FAIL (aliasing)\n\n");
flint_printf("regularized = %d\n\n", regularized);
flint_printf("S = "); acb_poly_printd(S, 15); flint_printf("\n\n");
flint_printf("A = "); acb_poly_printd(A, 15); flint_printf("\n\n");
flint_abort();
}
acb_poly_clear(S);
acb_poly_clear(A);
acb_poly_clear(B);
acb_poly_clear(C);
acb_clear(s);
acb_clear(t);
acb_clear(c);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
long iter;
flint_rand_t state;
printf("evaluate....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 10000; iter++)
{
long qbits1, qbits2, rbits1, rbits2, rbits3;
fmpq_poly_t F;
fmpq_t X, Y;
acb_poly_t f;
acb_t x, y;
qbits1 = 2 + n_randint(state, 200);
qbits2 = 2 + n_randint(state, 200);
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
rbits3 = 2 + n_randint(state, 200);
fmpq_poly_init(F);
fmpq_init(X);
fmpq_init(Y);
acb_poly_init(f);
acb_init(x);
acb_init(y);
fmpq_poly_randtest(F, state, 1 + n_randint(state, 20), qbits1);
fmpq_randtest(X, state, qbits2);
fmpq_poly_evaluate_fmpq(Y, F, X);
acb_poly_set_fmpq_poly(f, F, rbits1);
acb_set_fmpq(x, X, rbits2);
acb_poly_evaluate(y, f, x, rbits3);
if (!acb_contains_fmpq(y, Y))
{
printf("FAIL\n\n");
printf("F = "); fmpq_poly_print(F); printf("\n\n");
printf("X = "); fmpq_print(X); printf("\n\n");
printf("Y = "); fmpq_print(Y); printf("\n\n");
printf("f = "); acb_poly_printd(f, 15); printf("\n\n");
printf("x = "); acb_printd(x, 15); printf("\n\n");
printf("y = "); acb_printd(y, 15); printf("\n\n");
abort();
}
/* aliasing */
acb_poly_evaluate(x, f, x, rbits3);
if (!acb_contains_fmpq(x, Y))
{
printf("FAIL (aliasing)\n\n");
abort();
}
fmpq_poly_clear(F);
fmpq_clear(X);
fmpq_clear(Y);
acb_poly_clear(f);
acb_clear(x);
acb_clear(y);
}
flint_randclear(state);
flint_cleanup();
printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("divrem....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 100000; iter++)
{
slong m, n, qbits1, qbits2, rbits1, rbits2, rbits3;
fmpq_poly_t A, B, Q, R;
acb_poly_t a, b, q, r;
qbits1 = 2 + n_randint(state, 200);
qbits2 = 2 + n_randint(state, 200);
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
rbits3 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 20);
n = 1 + n_randint(state, 20);
fmpq_poly_init(A);
fmpq_poly_init(B);
fmpq_poly_init(Q);
fmpq_poly_init(R);
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(q);
acb_poly_init(r);
fmpq_poly_randtest(A, state, m, qbits1);
fmpq_poly_randtest_not_zero(B, state, n, qbits2);
fmpq_poly_divrem(Q, R, A, B);
acb_poly_set_fmpq_poly(a, A, rbits1);
acb_poly_set_fmpq_poly(b, B, rbits2);
acb_poly_divrem(q, r, a, b, rbits3);
if (!acb_poly_contains_fmpq_poly(q, Q) ||
!acb_poly_contains_fmpq_poly(r, R))
{
flint_printf("FAIL\n\n");
flint_printf("A = "); fmpq_poly_print(A); flint_printf("\n\n");
flint_printf("B = "); fmpq_poly_print(B); flint_printf("\n\n");
flint_printf("Q = "); fmpq_poly_print(Q); flint_printf("\n\n");
flint_printf("R = "); fmpq_poly_print(R); flint_printf("\n\n");
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("q = "); acb_poly_printd(q, 15); flint_printf("\n\n");
flint_printf("r = "); acb_poly_printd(r, 15); flint_printf("\n\n");
abort();
}
acb_poly_divrem(a, r, a, b, rbits3);
if (!acb_poly_equal(a, q))
{
flint_printf("FAIL (aliasing q, a)\n\n");
}
acb_poly_set_fmpq_poly(a, A, rbits1);
acb_poly_divrem(b, r, a, b, rbits3);
if (!acb_poly_equal(b, q))
{
flint_printf("FAIL (aliasing q, b)\n\n");
abort();
}
acb_poly_set_fmpq_poly(b, B, rbits2);
acb_poly_divrem(q, a, a, b, rbits3);
if (!acb_poly_equal(a, r))
{
flint_printf("FAIL (aliasing r, a)\n\n");
abort();
}
acb_poly_set_fmpq_poly(a, A, rbits1);
acb_poly_divrem(q, b, a, b, rbits3);
if (!acb_poly_equal(b, r))
{
flint_printf("FAIL (aliasing r, b)\n\n");
abort();
}
fmpq_poly_clear(A);
fmpq_poly_clear(B);
fmpq_poly_clear(Q);
fmpq_poly_clear(R);
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(q);
acb_poly_clear(r);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
long iter;
flint_rand_t state;
printf("interpolate_fast....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 10000; iter++)
{
long i, n, qbits1, qbits2, rbits1, rbits2, rbits3;
fmpq_poly_t P;
acb_poly_t R, S;
fmpq_t t, u;
acb_ptr xs, ys;
fmpq_poly_init(P);
acb_poly_init(R);
acb_poly_init(S);
fmpq_init(t);
fmpq_init(u);
qbits1 = 2 + n_randint(state, 200);
qbits2 = 2 + n_randint(state, 5);
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
rbits3 = 2 + n_randint(state, 200);
fmpq_poly_randtest(P, state, 1 + n_randint(state, 20), qbits1);
n = P->length;
xs = _acb_vec_init(n);
ys = _acb_vec_init(n);
acb_poly_set_fmpq_poly(R, P, rbits1);
if (n > 0)
{
fmpq_randtest(t, state, qbits2);
acb_set_fmpq(xs, t, rbits2);
for (i = 1; i < n; i++)
{
fmpq_randtest_not_zero(u, state, qbits2);
fmpq_abs(u, u);
fmpq_add(t, t, u);
acb_set_fmpq(xs + i, t, rbits2);
}
}
for (i = 0; i < n; i++)
acb_poly_evaluate(ys + i, R, xs + i, rbits2);
acb_poly_interpolate_fast(S, xs, ys, n, rbits3);
if (!acb_poly_contains_fmpq_poly(S, P))
{
printf("FAIL:\n");
printf("P = "); fmpq_poly_print(P); printf("\n\n");
printf("R = "); acb_poly_printd(R, 15); printf("\n\n");
printf("S = "); acb_poly_printd(S, 15); printf("\n\n");
abort();
}
fmpq_poly_clear(P);
acb_poly_clear(R);
acb_poly_clear(S);
fmpq_clear(t);
fmpq_clear(u);
_acb_vec_clear(xs, n);
_acb_vec_clear(ys, n);
}
flint_randclear(state);
flint_cleanup();
printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("airy_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 1000 * arb_test_multiplier(); iter++)
{
acb_poly_t ai, aip, bi, bip, ai2, aip2, bi2, bip2, z, w, t;
acb_t c;
slong n1, n2, prec1, prec2;
unsigned int mask;
acb_poly_init(ai); acb_poly_init(aip);
acb_poly_init(bi); acb_poly_init(bip);
acb_poly_init(ai2); acb_poly_init(aip2);
acb_poly_init(bi2); acb_poly_init(bip2);
acb_poly_init(z); acb_poly_init(w);
acb_poly_init(t); acb_init(c);
prec1 = 2 + n_randint(state, 300);
prec2 = 2 + n_randint(state, 300);
n1 = n_randint(state, 6);
n2 = n_randint(state, 6);
acb_poly_randtest(ai, state, 10, prec1, 10);
acb_poly_randtest(aip, state, 10, prec1, 10);
acb_poly_randtest(bi, state, 10, prec1, 10);
acb_poly_randtest(bip, state, 10, prec1, 10);
acb_poly_randtest(z, state, 1 + n_randint(state, 10), prec1, 10);
acb_hypgeom_airy_series(ai, aip, bi, bip, z, n1, prec1);
acb_poly_mullow(w, ai, bip, n1, prec1);
acb_poly_mullow(t, bi, aip, n1, prec1);
acb_poly_sub(w, w, t, prec1);
acb_const_pi(c, prec1);
acb_inv(c, c, prec1);
acb_poly_set_acb(t, c);
acb_poly_truncate(t, n1);
if (!acb_poly_overlaps(w, t))
{
flint_printf("FAIL: wronskian\n\n");
flint_printf("z = "); acb_poly_printd(z, 30); flint_printf("\n\n");
flint_printf("ai = "); acb_poly_printd(ai, 30); flint_printf("\n\n");
flint_printf("aip = "); acb_poly_printd(aip, 30); flint_printf("\n\n");
flint_printf("bi = "); acb_poly_printd(bi, 30); flint_printf("\n\n");
flint_printf("bip = "); acb_poly_printd(bip, 30); flint_printf("\n\n");
flint_printf("w = "); acb_poly_printd(w, 30); flint_printf("\n\n");
abort();
}
mask = n_randlimb(state);
acb_hypgeom_airy_series((mask & 1) ? ai2 : NULL,
(mask & 2) ? aip2 : NULL,
(mask & 4) ? bi2 : NULL,
(mask & 8) ? bip2 : NULL, z, n2, prec2);
acb_poly_truncate(ai, FLINT_MIN(n1, n2));
acb_poly_truncate(aip, FLINT_MIN(n1, n2));
acb_poly_truncate(bi, FLINT_MIN(n1, n2));
acb_poly_truncate(bip, FLINT_MIN(n1, n2));
acb_poly_truncate(ai2, FLINT_MIN(n1, n2));
acb_poly_truncate(aip2, FLINT_MIN(n1, n2));
acb_poly_truncate(bi2, FLINT_MIN(n1, n2));
acb_poly_truncate(bip2, FLINT_MIN(n1, n2));
if (((mask & 1) && (!acb_poly_overlaps(ai, ai2))) ||
((mask & 2) && (!acb_poly_overlaps(aip, aip2))) ||
((mask & 4) && (!acb_poly_overlaps(bi, bi2))) ||
((mask & 8) && (!acb_poly_overlaps(bip, bip2))))
{
flint_printf("FAIL: consistency (mask)\n\n");
flint_printf("mask = %u\n\n", mask);
flint_printf("len1 = %wd, len2 = %wd\n\n", n1, n2);
flint_printf("z = "); acb_poly_printd(z, 30); flint_printf("\n\n");
flint_printf("ai = "); acb_poly_printd(ai, 30); flint_printf("\n\n");
flint_printf("ai2 = "); acb_poly_printd(ai2, 30); flint_printf("\n\n");
flint_printf("aip = "); acb_poly_printd(aip, 30); flint_printf("\n\n");
flint_printf("aip2 = "); acb_poly_printd(aip2, 30); flint_printf("\n\n");
flint_printf("bi = "); acb_poly_printd(bi, 30); flint_printf("\n\n");
flint_printf("bi2 = "); acb_poly_printd(bi2, 30); flint_printf("\n\n");
flint_printf("bip = "); acb_poly_printd(bip, 30); flint_printf("\n\n");
flint_printf("bip2 = "); acb_poly_printd(bip2, 30); flint_printf("\n\n");
abort();
}
acb_poly_clear(ai); acb_poly_clear(aip);
acb_poly_clear(bi); acb_poly_clear(bip);
acb_poly_clear(ai2); acb_poly_clear(aip2);
acb_poly_clear(bi2); acb_poly_clear(bip2);
acb_poly_clear(z); acb_poly_clear(w);
acb_poly_clear(t); acb_clear(c);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("sqrt_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 5000; iter++)
{
slong m, n, qbits, rbits1, rbits2;
fmpq_poly_t A;
acb_poly_t a, b, c;
qbits = 2 + n_randint(state, 200);
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 30);
n = 1 + n_randint(state, 30);
fmpq_poly_init(A);
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
fmpq_poly_randtest_not_zero(A, state, m, qbits);
acb_poly_set_fmpq_poly(a, A, rbits1);
acb_poly_sqrt_series(b, a, n, rbits2);
/* Check sqrt(a)^2 = a */
acb_poly_mullow(c, b, b, n, rbits2);
fmpq_poly_truncate(A, n);
if (!acb_poly_contains_fmpq_poly(c, A))
{
flint_printf("FAIL\n\n");
flint_printf("bits2 = %wd\n", rbits2);
flint_printf("A = "); fmpq_poly_print(A); flint_printf("\n\n");
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("c = "); acb_poly_printd(c, 15); flint_printf("\n\n");
abort();
}
acb_poly_sqrt_series(a, a, n, rbits2);
if (!acb_poly_equal(a, b))
{
flint_printf("FAIL (aliasing)\n\n");
abort();
}
fmpq_poly_clear(A);
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("cos_pi_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 1000 * arb_test_multiplier(); iter++)
{
slong m, n1, n2, bits1, bits2, bits3;
acb_poly_t S, A, B, C;
acb_t pi;
bits1 = 2 + n_randint(state, 200);
bits2 = 2 + n_randint(state, 200);
bits3 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 30);
n1 = 1 + n_randint(state, 30);
n2 = 1 + n_randint(state, 30);
acb_poly_init(S);
acb_poly_init(A);
acb_poly_init(B);
acb_poly_init(C);
acb_init(pi);
acb_poly_randtest(S, state, m, bits1, 3);
acb_poly_randtest(A, state, m, bits1, 3);
acb_poly_randtest(B, state, m, bits1, 3);
acb_poly_cos_pi_series(A, S, n1, bits2);
acb_const_pi(pi, bits3);
acb_poly_set_acb(B, pi);
acb_poly_mul(B, S, B, bits3);
acb_poly_cos_series(B, B, n2, bits3);
acb_poly_set(C, A);
acb_poly_truncate(C, FLINT_MIN(n1, n2));
acb_poly_truncate(B, FLINT_MIN(n1, n2));
if (!acb_poly_overlaps(B, C))
{
flint_printf("FAIL\n\n");
flint_printf("S = "); acb_poly_printd(S, 15); flint_printf("\n\n");
flint_printf("A = "); acb_poly_printd(A, 15); flint_printf("\n\n");
flint_printf("B = "); acb_poly_printd(B, 15); flint_printf("\n\n");
abort();
}
acb_poly_cos_pi_series(S, S, n1, bits2);
if (!acb_poly_overlaps(A, S))
{
flint_printf("FAIL (aliasing)\n\n");
abort();
}
acb_poly_clear(S);
acb_poly_clear(A);
acb_poly_clear(B);
acb_poly_clear(C);
acb_clear(pi);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("sin_cos_series_basecase....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 1000 * arb_test_multiplier(); iter++)
{
slong m, n, rbits1, rbits2;
fmpq_poly_t B;
acb_poly_t a, b, c, d, e;
int times_pi;
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 30);
n = 1 + n_randint(state, 30);
times_pi = n_randint(state, 2);
fmpq_poly_init(B);
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
acb_poly_init(d);
acb_poly_init(e);
acb_poly_randtest(a, state, m, rbits1, 10);
acb_poly_sin_cos_series_basecase(b, c, a, n, rbits2, times_pi);
/* Check sin(x)^2 + cos(x)^2 = 1 */
acb_poly_mullow(d, b, b, n, rbits2);
acb_poly_mullow(e, c, c, n, rbits2);
acb_poly_add(d, d, e, rbits2);
fmpq_poly_one(B);
if (!acb_poly_contains_fmpq_poly(d, B))
{
flint_printf("FAIL\n\n");
flint_printf("bits2 = %wd\n", rbits2);
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("c = "); acb_poly_printd(c, 15); flint_printf("\n\n");
flint_printf("d = "); acb_poly_printd(d, 15); flint_printf("\n\n");
abort();
}
acb_poly_set(d, a);
acb_poly_sin_cos_series_basecase(d, c, d, n, rbits2, times_pi);
if (!acb_poly_equal(b, d))
{
flint_printf("FAIL (aliasing 1)\n\n");
abort();
}
acb_poly_set(d, a);
acb_poly_sin_cos_series_basecase(b, d, d, n, rbits2, times_pi);
if (!acb_poly_equal(c, d))
{
flint_printf("FAIL (aliasing 2)\n\n");
abort();
}
fmpq_poly_clear(B);
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
acb_poly_clear(d);
acb_poly_clear(e);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("pow_ui_trunc_binexp....");
fflush(stdout);
flint_randinit(state);
/* compare with fmpz_poly */
for (iter = 0; iter < 10000 * arb_test_multiplier(); iter++)
{
slong zbits1, rbits1, rbits2, trunc;
ulong e;
fmpz_poly_t A, B;
acb_poly_t a, b;
zbits1 = 2 + n_randint(state, 100);
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
e = n_randint(state, 50);
trunc = n_randint(state, 40);
fmpz_poly_init(A);
fmpz_poly_init(B);
acb_poly_init(a);
acb_poly_init(b);
fmpz_poly_randtest(A, state, 1 + n_randint(state, 10), zbits1);
fmpz_poly_pow_trunc(B, A, e, trunc);
acb_poly_set_fmpz_poly(a, A, rbits1);
acb_poly_pow_ui_trunc_binexp(b, a, e, trunc, rbits2);
if (!acb_poly_contains_fmpz_poly(b, B))
{
flint_printf("FAIL\n\n");
flint_printf("bits2 = %wd\n", rbits2);
flint_printf("e = %wu\n", e);
flint_printf("trunc = %wd\n", trunc);
flint_printf("A = "); fmpz_poly_print(A); flint_printf("\n\n");
flint_printf("B = "); fmpz_poly_print(B); flint_printf("\n\n");
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
abort();
}
acb_poly_pow_ui_trunc_binexp(a, a, e, trunc, rbits2);
if (!acb_poly_equal(a, b))
{
flint_printf("FAIL (aliasing)\n\n");
abort();
}
fmpz_poly_clear(A);
fmpz_poly_clear(B);
acb_poly_clear(a);
acb_poly_clear(b);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int
main(void)
{
int iter;
flint_rand_t state;
flint_printf("set_trunc_round....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 1000 * arb_test_multiplier(); iter++)
{
acb_poly_t a, b, c, d, e;
slong n, prec;
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
acb_poly_init(d);
acb_poly_init(e);
acb_poly_randtest(a, state, n_randint(state, 10), 2 + n_randint(state, 200), 10);
acb_poly_randtest(b, state, n_randint(state, 10), 2 + n_randint(state, 200), 10);
acb_poly_randtest(c, state, n_randint(state, 10), 2 + n_randint(state, 200), 10);
acb_poly_randtest(d, state, n_randint(state, 10), 2 + n_randint(state, 200), 10);
acb_poly_randtest(e, state, n_randint(state, 10), 2 + n_randint(state, 200), 10);
n = n_randint(state, 10);
prec = 2 + n_randint(state, 200);
acb_poly_set_trunc(b, a, n);
acb_poly_set_round(b, b, prec);
acb_poly_set_round(c, a, prec);
acb_poly_set_trunc(c, c, n);
acb_poly_set_trunc_round(d, a, n, prec);
acb_poly_set(e, a);
acb_poly_set_trunc_round(e, e, n, prec);
if (!acb_poly_equal(b, c) || !acb_poly_equal(c, d) || !acb_poly_equal(d, e))
{
flint_printf("FAIL\n\n");
acb_poly_printd(a, 50), flint_printf("\n\n");
acb_poly_printd(b, 50), flint_printf("\n\n");
acb_poly_printd(c, 50), flint_printf("\n\n");
acb_poly_printd(d, 50), flint_printf("\n\n");
acb_poly_printd(e, 50), flint_printf("\n\n");
flint_abort();
}
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
acb_poly_clear(d);
acb_poly_clear(e);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return 0;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("mullow_transpose....");
fflush(stdout);
flint_randinit(state);
/* compare with fmpq_poly */
for (iter = 0; iter < 10000 * arb_test_multiplier(); iter++)
{
slong qbits1, qbits2, rbits1, rbits2, rbits3, trunc;
fmpq_poly_t A, B, C;
acb_poly_t a, b, c, d;
qbits1 = 2 + n_randint(state, 200);
qbits2 = 2 + n_randint(state, 200);
rbits1 = 2 + n_randint(state, 200);
rbits2 = 2 + n_randint(state, 200);
rbits3 = 2 + n_randint(state, 200);
trunc = n_randint(state, 10);
fmpq_poly_init(A);
fmpq_poly_init(B);
fmpq_poly_init(C);
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
acb_poly_init(d);
fmpq_poly_randtest(A, state, 1 + n_randint(state, 10), qbits1);
fmpq_poly_randtest(B, state, 1 + n_randint(state, 10), qbits2);
fmpq_poly_mullow(C, A, B, trunc);
acb_poly_set_fmpq_poly(a, A, rbits1);
acb_poly_set_fmpq_poly(b, B, rbits2);
acb_poly_mullow_transpose(c, a, b, trunc, rbits3);
if (!acb_poly_contains_fmpq_poly(c, C))
{
flint_printf("FAIL\n\n");
flint_printf("bits3 = %wd\n", rbits3);
flint_printf("trunc = %wd\n", trunc);
flint_printf("A = "); fmpq_poly_print(A); flint_printf("\n\n");
flint_printf("B = "); fmpq_poly_print(B); flint_printf("\n\n");
flint_printf("C = "); fmpq_poly_print(C); flint_printf("\n\n");
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("c = "); acb_poly_printd(c, 15); flint_printf("\n\n");
abort();
}
acb_poly_set(d, a);
acb_poly_mullow_transpose(d, d, b, trunc, rbits3);
if (!acb_poly_equal(d, c))
{
flint_printf("FAIL (aliasing 1)\n\n");
abort();
}
acb_poly_set(d, b);
acb_poly_mullow_transpose(d, a, d, trunc, rbits3);
if (!acb_poly_equal(d, c))
{
flint_printf("FAIL (aliasing 2)\n\n");
abort();
}
/* test squaring */
acb_poly_set(b, a);
acb_poly_mullow_transpose(c, a, b, trunc, rbits3);
acb_poly_mullow_transpose(d, a, a, trunc, rbits3);
if (!acb_poly_overlaps(c, d)) /* not guaranteed to be identical */
{
flint_printf("FAIL (squaring)\n\n");
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("c = "); acb_poly_printd(c, 15); flint_printf("\n\n");
abort();
}
acb_poly_mullow_transpose(a, a, a, trunc, rbits3);
if (!acb_poly_equal(d, a))
{
flint_printf("FAIL (aliasing, squaring)\n\n");
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("d = "); acb_poly_printd(d, 15); flint_printf("\n\n");
abort();
}
fmpq_poly_clear(A);
fmpq_poly_clear(B);
fmpq_poly_clear(C);
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
acb_poly_clear(d);
}
/* compare with classical */
for (iter = 0; iter < 10000 * arb_test_multiplier(); iter++)
{
slong bits, trunc;
acb_poly_t a, b, ab, ab2;
bits = 2 + n_randint(state, 200);
trunc = n_randint(state, 10);
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(ab);
acb_poly_init(ab2);
acb_poly_randtest(a, state, 1 + n_randint(state, 10), bits, 5);
acb_poly_randtest(b, state, 1 + n_randint(state, 10), bits, 5);
acb_poly_mullow_classical(ab, a, b, trunc, bits);
acb_poly_mullow_transpose(ab2, a, b, trunc, bits);
if (!acb_poly_overlaps(ab, ab2))
{
flint_printf("FAIL\n\n");
flint_printf("bits = %wd\n", bits);
flint_printf("trunc = %wd\n", trunc);
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("ab = "); acb_poly_printd(ab, 15); flint_printf("\n\n");
flint_printf("ab2 = "); acb_poly_printd(ab2, 15); flint_printf("\n\n");
abort();
}
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(ab);
acb_poly_clear(ab2);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("digamma_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 1000 * arb_test_multiplier(); iter++)
{
slong m, n1, n2, rbits1, rbits2, rbits3;
acb_poly_t a, b, c, d;
rbits1 = 2 + n_randint(state, 300);
rbits2 = 2 + n_randint(state, 300);
rbits3 = 2 + n_randint(state, 300);
m = n_randint(state, 25);
n1 = n_randint(state, 25);
n2 = n_randint(state, 25);
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
acb_poly_init(d);
acb_poly_randtest(a, state, m, rbits1, 10);
acb_poly_digamma_series(b, a, n1, rbits2);
acb_poly_digamma_series(c, a, n2, rbits3);
acb_poly_set(d, b);
acb_poly_truncate(d, FLINT_MIN(n1, n2));
acb_poly_truncate(c, FLINT_MIN(n1, n2));
if (!acb_poly_overlaps(c, d))
{
flint_printf("FAIL\n\n");
flint_printf("n1 = %wd, n2 = %wd, bits2 = %wd, bits3 = %wd\n", n1, n2, rbits2, rbits3);
flint_printf("a = "); acb_poly_printd(a, 50); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 50); flint_printf("\n\n");
flint_printf("c = "); acb_poly_printd(c, 50); flint_printf("\n\n");
flint_abort();
}
/* check psi(a) + 1/a = psi(a+1) */
acb_poly_inv_series(c, a, n1, rbits2);
acb_poly_add(c, b, c, rbits2);
acb_poly_add_si(d, a, 1, rbits2);
acb_poly_digamma_series(d, d, n1, rbits2);
if (!acb_poly_overlaps(c, d))
{
flint_printf("FAIL (functional equation)\n\n");
flint_printf("a = "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b = "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("c = "); acb_poly_printd(c, 15); flint_printf("\n\n");
flint_printf("d = "); acb_poly_printd(d, 15); flint_printf("\n\n");
flint_abort();
}
acb_poly_digamma_series(a, a, n1, rbits2);
if (!acb_poly_overlaps(a, b))
{
flint_printf("FAIL (aliasing)\n\n");
flint_abort();
}
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
acb_poly_clear(d);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("li_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 200 * arb_test_multiplier(); iter++)
{
slong m, n1, n2, n3, bits1, bits2, bits3;
acb_poly_t S, A, B, C, T, U;
int offset;
bits1 = 2 + n_randint(state, 200);
bits2 = 2 + n_randint(state, 200);
bits3 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 10);
n1 = 1 + n_randint(state, 10);
n2 = 1 + n_randint(state, 10);
n3 = FLINT_MIN(n1, n2);
offset = n_randint(state, 2);
acb_poly_init(S);
acb_poly_init(A);
acb_poly_init(B);
acb_poly_init(C);
acb_poly_init(T);
acb_poly_init(U);
acb_poly_randtest(S, state, m, bits1, 3);
acb_poly_randtest(A, state, m, bits1, 3);
acb_poly_randtest(B, state, m, bits1, 3);
acb_hypgeom_li_series(A, S, offset, n1, bits2);
acb_hypgeom_li_series(B, S, offset, n2, bits3);
acb_poly_set(C, A);
acb_poly_truncate(C, n3);
acb_poly_truncate(B, n3);
/* [li(h(x))]' log(h(x)) = h'(x) */
acb_poly_derivative(T, A, bits2);
acb_poly_log_series(U, S, n3, bits2);
acb_poly_mullow(T, T, U, FLINT_MAX(0, n3 - 1), bits2);
acb_poly_derivative(U, S, bits2);
acb_poly_truncate(U, FLINT_MAX(0, n3 - 1));
if (!acb_poly_overlaps(B, C) || !acb_poly_overlaps(T, U))
{
flint_printf("FAIL\n\n");
flint_printf("S = "); acb_poly_printd(S, 15); flint_printf("\n\n");
flint_printf("A = "); acb_poly_printd(A, 15); flint_printf("\n\n");
flint_printf("B = "); acb_poly_printd(B, 15); flint_printf("\n\n");
flint_printf("T = "); acb_poly_printd(T, 15); flint_printf("\n\n");
flint_printf("U = "); acb_poly_printd(U, 15); flint_printf("\n\n");
abort();
}
acb_hypgeom_li_series(S, S, offset, n1, bits2);
if (!acb_poly_overlaps(A, S))
{
flint_printf("FAIL (aliasing)\n\n");
abort();
}
acb_poly_clear(S);
acb_poly_clear(A);
acb_poly_clear(B);
acb_poly_clear(C);
acb_poly_clear(T);
acb_poly_clear(U);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("binomial_transform....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 500 * arb_test_multiplier(); iter++)
{
acb_poly_t a, b, c, d;
slong j, n, prec;
acb_poly_init(a);
acb_poly_init(b);
acb_poly_init(c);
acb_poly_init(d);
n = n_randint(state, 20);
prec = 2 + n_randint(state, 200);
acb_poly_randtest(a, state, n, prec, 10);
acb_poly_randtest(b, state, n, prec, 10);
acb_poly_randtest(c, state, n, prec, 10);
/* check self-inversion property */
acb_poly_binomial_transform(b, a, n, prec);
acb_poly_binomial_transform(c, b, n, prec);
acb_poly_set(d, a);
acb_poly_truncate(d, n);
if (!acb_poly_contains(c, d))
{
flint_printf("FAIL (containment)\n\n");
flint_printf("n = %wd, prec = %wd\n\n", n, prec);
flint_printf("a: "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b: "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("c: "); acb_poly_printd(c, 15); flint_printf("\n\n");
flint_printf("d: "); acb_poly_printd(d, 15); flint_printf("\n\n");
abort();
}
acb_poly_set(d, a);
acb_poly_binomial_transform(d, d, n, prec);
if (!acb_poly_equal(d, b))
{
flint_printf("FAIL (aliasing)\n\n");
flint_printf("a: "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b: "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("d: "); acb_poly_printd(d, 15); flint_printf("\n\n");
abort();
}
/* compare with power series operations */
acb_poly_zero(d);
for (j = 1; j < n; j++)
acb_poly_set_coeff_si(d, j, -1);
acb_poly_compose_series(c, a, d, n, prec);
for (j = 0; j < n; j++)
acb_poly_set_coeff_si(d, j, 1);
acb_poly_mullow(c, c, d, n, prec);
if (!acb_poly_overlaps(b, c))
{
flint_printf("FAIL (power series)\n\n");
flint_printf("a: "); acb_poly_printd(a, 15); flint_printf("\n\n");
flint_printf("b: "); acb_poly_printd(b, 15); flint_printf("\n\n");
flint_printf("c: "); acb_poly_printd(c, 15); flint_printf("\n\n");
abort();
}
acb_poly_clear(a);
acb_poly_clear(b);
acb_poly_clear(c);
acb_poly_clear(d);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
