arb/acb_hypgeom/test/t-u_asymp.c
Tommy Hofmann 6bf072eb59 Replace abort with flint_abort.
This will allow us to not loose the julia session on error.
See also https://github.com/wbhart/flint2/pull/243
2017-02-28 16:52:57 +01:00

216 lines
7 KiB
C

/*
Copyright (C) 2014 Fredrik Johansson
This file is part of Arb.
Arb is free software: you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License (LGPL) as published
by the Free Software Foundation; either version 2.1 of the License, or
(at your option) any later version. See <http://www.gnu.org/licenses/>.
*/
#include "acb_hypgeom.h"
int main()
{
slong iter;
flint_rand_t state;
flint_printf("u_asymp....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 10000 * arb_test_multiplier(); iter++)
{
acb_t a, b, a2, b2, z, U1, U2, t, u, M1, M2, am;
acb_struct bm[2];
ulong n1, n2;
slong prec0, prec1, prec2;
acb_init(a);
acb_init(b);
acb_init(a2);
acb_init(b2);
acb_init(z);
acb_init(U1);
acb_init(U2);
acb_init(t);
acb_init(u);
acb_init(M1);
acb_init(M2);
acb_init(am);
acb_init(bm);
acb_init(bm + 1);
if (n_randint(state, 4) == 0)
{
n1 = n_randint(state, 200);
n2 = n_randint(state, 200);
prec0 = 2 + n_randint(state, 1000);
prec1 = 2 + n_randint(state, 1000);
prec2 = 2 + n_randint(state, 1000);
}
else
{
n1 = n_randint(state, 40);
n2 = n_randint(state, 40);
prec0 = 2 + n_randint(state, 300);
prec1 = 2 + n_randint(state, 300);
prec2 = 2 + n_randint(state, 300);
}
acb_randtest_param(a, state, prec0, 1 + n_randint(state, 20));
if (n_randint(state, 4) == 0)
acb_add_ui(b, a, n_randint(state, 10), prec0);
else
acb_randtest_param(b, state, prec0, 1 + n_randint(state, 20));
acb_randtest(z, state, prec0, 1 + n_randint(state, 20));
/* Test Kummer's transformation */
acb_sub(a2, a, b, prec0);
acb_add_ui(a2, a2, 1, prec0);
acb_sub_ui(b2, b, 2, prec0);
acb_neg(b2, b2);
acb_hypgeom_u_asymp(U1, a, b, z, n1, prec1);
acb_hypgeom_u_asymp(U2, a2, b2, z, n2, prec2);
if (!acb_overlaps(U1, U2))
{
flint_printf("FAIL (Kummer transformation)\n");
flint_printf("iter = %wd\n", iter);
flint_printf("a = "); acb_printd(a, 50); flint_printf("\n");
flint_printf("b = "); acb_printd(b, 50); flint_printf("\n");
flint_printf("z = "); acb_printd(z, 50); flint_printf("\n");
flint_printf("n1 = %wd, n2 = %wd, prec1 = %wd, prec2 = %wd\n", n1, n2, prec1, prec2);
flint_printf("U1 = "); acb_printd(U1, 100); flint_printf("\n");
flint_printf("U2 = "); acb_printd(U2, 100); flint_printf("\n");
flint_abort();
}
/* Check contiguous relation
(b-a-1)U(a,b-1,z) + z U(a,b+1,z) + (1-b-z) U(a,b,z) = 0 */
acb_one(t);
acb_sub(t, t, b, prec0);
acb_sub(t, t, z, prec0);
acb_mul(u, U1, t, prec1);
acb_add_ui(b2, b, 1, prec0);
acb_hypgeom_u_asymp(U2, a, b2, z, n2, prec2);
acb_addmul(u, U2, z, prec1);
acb_sub_ui(b2, b, 1, prec0);
acb_hypgeom_u_asymp(U2, a, b2, z, n2, prec2);
acb_sub(t, b, a, prec0);
acb_sub_ui(t, t, 1, prec0);
acb_mul(t, t, U2, prec0);
acb_add(t, t, u, prec0);
if (!acb_contains_zero(t))
{
flint_printf("FAIL (contiguous relation)\n");
flint_printf("iter = %wd\n", iter);
flint_printf("a = "); acb_printd(a, 50); flint_printf("\n");
flint_printf("b = "); acb_printd(b, 50); flint_printf("\n");
flint_printf("z = "); acb_printd(z, 50); flint_printf("\n");
flint_printf("n1 = %wd, n2 = %wd, prec1 = %wd, prec2 = %wd\n", n1, n2, prec1, prec2);
flint_printf("U1 = "); acb_printd(U1, 100); flint_printf("\n");
flint_printf("t = "); acb_printd(t, 100); flint_printf("\n");
flint_abort();
}
/* Test U(a,b,z) = gamma(1-b)/gamma(a-b+1) M(a,b,z)
+ gamma(b-1)/gamma(a) z^(1-b) M(a-b+1,2-b,z) */
acb_set(am, a);
acb_set(bm, b);
acb_one(bm + 1);
acb_hypgeom_pfq_direct(M1, am, 1, bm, 2, z, n2, prec2);
acb_sub(am, a, b, prec2);
acb_add_ui(am, am, 1, prec2);
acb_sub_ui(bm, b, 2, prec2);
acb_neg(bm, bm);
acb_one(bm + 1);
acb_hypgeom_pfq_direct(M2, am, 1, bm, 2, z, n2, prec2);
acb_sub(am, a, b, prec2);
acb_add_ui(am, am, 1, prec2);
acb_rgamma(am, am, prec2);
acb_mul(M1, M1, am, prec2);
acb_sub_ui(am, b, 1, prec2);
acb_neg(am, am);
acb_gamma(am, am, prec2);
acb_mul(M1, M1, am, prec2);
acb_rgamma(am, a, prec2);
acb_mul(M2, M2, am, prec2);
acb_sub_ui(am, b, 1, prec2);
acb_gamma(am, am, prec2);
acb_mul(M2, M2, am, prec2);
acb_sub_ui(am, b, 1, prec2);
acb_neg(am, am);
acb_pow(am, z, am, prec2);
acb_mul(M2, M2, am, prec2);
acb_add(U2, M1, M2, prec2);
acb_pow(am, z, a, prec2);
acb_mul(U2, U2, am, prec2);
if (!acb_overlaps(U1, U2))
{
flint_printf("FAIL (U in terms of M)\n");
flint_printf("iter = %wd\n", iter);
flint_printf("a = "); acb_printd(a, 50); flint_printf("\n");
flint_printf("b = "); acb_printd(b, 50); flint_printf("\n");
flint_printf("z = "); acb_printd(z, 50); flint_printf("\n");
flint_printf("n1 = %wd, n2 = %wd, prec1 = %wd, prec2 = %wd\n", n1, n2, prec1, prec2);
flint_printf("U1 = "); acb_printd(U1, 100); flint_printf("\n");
flint_printf("U2 = "); acb_printd(U2, 100); flint_printf("\n");
flint_abort();
}
/* Test special value: b = a+1 */
acb_add_ui(b, a, 1, prec0);
acb_hypgeom_u_asymp(U1, a, b, z, n1, prec1);
acb_one(U2);
if (!acb_overlaps(U1, U2))
{
flint_printf("FAIL (special value)\n");
flint_printf("iter = %wd\n", iter);
flint_printf("a = "); acb_printd(a, 50); flint_printf("\n");
flint_printf("b = "); acb_printd(b, 50); flint_printf("\n");
flint_printf("z = "); acb_printd(z, 50); flint_printf("\n");
flint_printf("n1 = %wd, n2 = %wd, prec1 = %wd, prec2 = %wd\n", n1, n2, prec1, prec2);
flint_printf("U1 = "); acb_printd(U1, 100); flint_printf("\n");
flint_printf("U2 = "); acb_printd(U2, 100); flint_printf("\n");
flint_abort();
}
acb_clear(a);
acb_clear(b);
acb_clear(a2);
acb_clear(b2);
acb_clear(z);
acb_clear(U1);
acb_clear(U2);
acb_clear(t);
acb_clear(u);
acb_clear(M1);
acb_clear(M2);
acb_clear(am);
acb_clear(bm);
acb_clear(bm + 1);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}