#include "test_helpers.h"
#include "ulong_extras.h"
#include "fmpz.h"
#include "fmpz_mod.h"
#include "fmpz_mod_poly.h"
TEST_FUNCTION_START(fmpz_mod_poly_find_distinct_nonzero_roots, state)
{
slong i, j;
fmpz_mod_ctx_t ctx;
fmpz_mod_ctx_init_ui(ctx, 2);
for (i = 0; i < 200 * flint_test_multiplier(); i++)
{
int highdegreefactor;
fmpz_mod_poly_t a, b, r;
fmpz_t p, e, zero;
fmpz_init(zero);
fmpz_init_set_ui(p, n_randtest_prime(state, 1));
fmpz_mod_ctx_set_modulus(ctx, p);
fmpz_init(e);
fmpz_mod_poly_init(a, ctx);
fmpz_mod_poly_init(b, ctx);
fmpz_mod_poly_init(r, ctx);
fmpz_mod_poly_set_ui(a, 1 + n_randint(state, fmpz_get_ui(p) - 1), ctx);
highdegreefactor = 0;
for (j = n_randint(state, 10); j >= 0; j--)
{
if (n_randint(state, 10) > 1)
{
fmpz_mod_poly_randtest_monic_irreducible(b, state, 1, ctx);
}
else
{
highdegreefactor = 1;
fmpz_mod_poly_randtest_monic_irreducible(b, state, 2 + n_randint(state, 9), ctx);
}
fmpz_mod_poly_mul(a, a, b, ctx);
}
fmpz_mod_poly_fit_length(r, fmpz_mod_poly_degree(a, ctx), ctx);
if (fmpz_mod_poly_find_distinct_nonzero_roots(r->coeffs, a, ctx))
{
fmpz_mod_poly_derivative(b, a, ctx);
fmpz_mod_poly_gcd(b, b, a, ctx);
if (fmpz_mod_poly_degree(b, ctx) > 0)
{
flint_printf("FAIL\ncheck multiple roots i = %wd\n", i);
fflush(stdout);
flint_abort();
}
for (j = fmpz_mod_poly_degree(a, ctx) - 1; j >= 0; j--)
{
if (fmpz_is_zero(r->coeffs + j))
{
flint_printf("FAIL\ncheck zero root i = %wd\n", i);
fflush(stdout);
flint_abort();
}
if (fmpz_mod_poly_evaluate_fmpz(e, a, r->coeffs + j, ctx), !fmpz_is_zero(e))
{
flint_printf("FAIL\ncheck root is a root i = %wd\n", i);
fflush(stdout);
flint_abort();
}
}
}
else
{
fmpz_mod_poly_derivative(b, a, ctx);
fmpz_mod_poly_gcd(b, b, a, ctx);
if (!highdegreefactor
&& fmpz_mod_poly_degree(b, ctx) == 0
&& (fmpz_mod_poly_evaluate_fmpz(e, a, zero, ctx), !fmpz_is_zero(e)))
{
flint_printf("FAIL\ncheck fail return i = %wd\n", i);
fflush(stdout);
flint_abort();
}
}
fmpz_mod_poly_clear(a, ctx);
fmpz_mod_poly_clear(b, ctx);
fmpz_mod_poly_clear(r, ctx);
fmpz_clear(p);
fmpz_clear(e);
fmpz_clear(zero);
}
fmpz_mod_ctx_clear(ctx);
TEST_FUNCTION_END(state);
}