#include "test_helpers.h"
#include "ulong_extras.h"
#include "long_extras.h"
#include "padic.h"
TEST_FUNCTION_START(padic_shift, state)
{
int i, result;
for (i = 0; i < 1000 * flint_test_multiplier(); i++)
{
fmpz_t p;
slong N;
padic_ctx_t ctx;
padic_t a, b, c;
slong v;
fmpz_init_set_ui(p, n_randtest_prime(state, 0));
N = n_randint(state, PADIC_TEST_PREC_MAX - PADIC_TEST_PREC_MIN)
+ PADIC_TEST_PREC_MIN;
padic_ctx_init(ctx, p, FLINT_MAX(0, N-10), FLINT_MAX(0, N+10), PADIC_SERIES);
padic_init2(a, N);
padic_init2(b, N);
padic_init2(c, N);
padic_randtest(a, state, ctx);
v = z_randint(state, (FLINT_ABS(N) + 4) / 3);
padic_set(b, a, ctx);
padic_shift(c, b, v, ctx);
padic_shift(b, b, v, ctx);
result = (padic_equal(b, c));
if (!result)
{
flint_printf("FAIL:\n\n");
flint_printf("a = "), padic_print(a, ctx), flint_printf("\n");
flint_printf("b = "), padic_print(b, ctx), flint_printf("\n");
flint_printf("c = "), padic_print(c, ctx), flint_printf("\n");
fflush(stdout);
flint_abort();
}
padic_clear(a);
padic_clear(b);
padic_clear(c);
fmpz_clear(p);
padic_ctx_clear(ctx);
}
for (i = 0; i < 1000 * flint_test_multiplier(); i++)
{
fmpz_t p;
slong N;
padic_ctx_t ctx;
padic_t a, b, c;
slong v, v1, v2;
fmpz_init_set_ui(p, n_randtest_prime(state, 0));
N = n_randint(state, PADIC_TEST_PREC_MAX - PADIC_TEST_PREC_MIN)
+ PADIC_TEST_PREC_MIN;
padic_ctx_init(ctx, p, FLINT_MAX(0, N-10), FLINT_MAX(0, N+10), PADIC_SERIES);
padic_init2(a, N);
padic_init2(b, N);
padic_init2(c, N);
padic_randtest(a, state, ctx);
v1 = z_randint(state, (FLINT_ABS(N) + 4) / 3);
v2 = z_randint(state, (FLINT_ABS(N) + 4) / 3);
padic_shift(b, a, v1, ctx);
padic_shift(b, b, v2, ctx);
padic_shift(c, a, v2, ctx);
padic_shift(c, c, v1, ctx);
v = FLINT_MIN(v1, v2);
v = FLINT_MIN(v, padic_val(a));
v = FLINT_MIN(v, 0);
if ((v >= 0) || (-v < N))
{
slong N2 = (v >= 0) ? N : N + v;
padic_prec(b) = N2;
padic_prec(c) = N2;
padic_reduce(b, ctx);
padic_reduce(c, ctx);
result = (padic_equal(b, c));
if (!result)
{
flint_printf("FAIL:\n\n");
flint_printf("a = "), padic_print(a, ctx), flint_printf("\n");
flint_printf("b = "), padic_print(b, ctx), flint_printf("\n");
flint_printf("c = "), padic_print(c, ctx), flint_printf("\n");
fflush(stdout);
flint_abort();
}
}
padic_clear(a);
padic_clear(b);
padic_clear(c);
fmpz_clear(p);
padic_ctx_clear(ctx);
}
TEST_FUNCTION_END(state);
}