#include "test_helpers.h"
#include "nmod_vec.h"
#include "nmod_poly.h"
#include "nmod.h"
TEST_FUNCTION_START(nmod_poly_evaluate_geometric_nmod_vec_fast, state)
{
int i, result = 1;
for (i = 0; i < 200 * flint_test_multiplier(); i++)
{
nmod_poly_t P;
nn_ptr y, z;
ulong mod, r;
slong n, npoints;
npoints = (i < 10) ? i : n_randint(state, 500);
n = n_randint(state, 1000);
do
{
mod = n_randtest_prime(state, 1);
}
while (mod <= 2*FLINT_MAX(npoints, n) + 1);
nmod_poly_init(P, mod);
r = n_primitive_root_prime(mod);
y = _nmod_vec_init(npoints);
z = _nmod_vec_init(npoints);
nmod_poly_randtest(P, state, n);
nmod_poly_evaluate_geometric_nmod_vec_iter(y, P, r, npoints);
nmod_poly_evaluate_geometric_nmod_vec_fast(z, P, r, npoints);
result = _nmod_vec_equal(y, z, npoints);
if (!result)
{
flint_printf("FAIL:\n");
flint_printf("mod=%wu, n=%wd, npoints=%wd\n\n", mod, n, npoints);
flint_printf("P: "); nmod_poly_print_pretty(P, "x"); flint_printf("\n\n");
flint_printf("y: "); _nmod_vec_print_pretty(y, npoints, P->mod); flint_printf("\n\n");
flint_printf("z: "); _nmod_vec_print_pretty(z, npoints, P->mod); flint_printf("\n\n");
fflush(stdout);
flint_abort();
}
nmod_poly_clear(P);
_nmod_vec_clear(y);
_nmod_vec_clear(z);
}
TEST_FUNCTION_END(state);
}