#include "test_helpers.h"
#include "ulong_extras.h"
#include "fmpz.h"
#include "nmod_mpoly_factor.h"
void check_content(const nmod_mpoly_t p, const nmod_mpoly_ctx_t ctx)
{
slong i, v;
nmod_mpoly_t q;
nmod_mpoly_factor_t g;
fmpz_t deg;
nmod_mpoly_factor_init(g, ctx);
nmod_mpoly_init(q, ctx);
fmpz_init(deg);
if (!nmod_mpoly_factor_content(g, p, ctx))
{
flint_printf("FAIL:\ncheck factorization could be computed\n");
fflush(stdout);
flint_abort();
}
for (i = 0; i < g->num; i++)
{
if (g->poly[i].length < 1 || g->poly[i].coeffs[0] != 1)
{
flint_printf("FAIL:\nfactorization is not unit normal\n");
fflush(stdout);
flint_abort();
}
}
nmod_mpoly_factor_expand(q, g, ctx);
if (!nmod_mpoly_equal(q, p, ctx))
{
flint_printf("FAIL:\nfactorization does not match original polynomial\n");
fflush(stdout);
flint_abort();
}
for (i = 0; i < g->num; i++)
{
for (v = 0; v < ctx->minfo->nvars; v++)
{
if (nmod_mpoly_length(g->poly + i, ctx) < 2)
{
if (!nmod_mpoly_is_gen(g->poly + i, -1, ctx))
{
flint_printf("FAIL:\nmonomial is bad\n");
fflush(stdout);
flint_abort();
}
}
else
{
if (!nmod_mpoly_content_vars(q, g->poly + i, &v, 1, ctx))
{
flint_printf("FAIL:\ncheck content could be computed\n");
fflush(stdout);
flint_abort();
}
nmod_mpoly_degree_fmpz(deg, g->poly + i, v, ctx);
if (!nmod_mpoly_is_one(q, ctx) && !fmpz_is_zero(deg))
{
flint_printf("FAIL:\ncontent is bad\n");
fflush(stdout);
flint_abort();
}
}
}
}
nmod_mpoly_clear(q, ctx);
nmod_mpoly_factor_clear(g, ctx);
fmpz_clear(deg);
}
TEST_FUNCTION_START(nmod_mpoly_factor_content, state)
{
slong i, j, k, tmul = 30;
for (i = 0; i < tmul * flint_test_multiplier(); i++)
{
nmod_mpoly_ctx_t ctx;
nmod_mpoly_t a, t;
slong n, nfacs, len;
ulong * expbounds;
ulong p;
p = n_randint(state, (i % 2 == 0) ? 4 : FLINT_BITS - 1) + 1;
p = n_randbits(state, p);
p = n_nextprime(p, 1);
nmod_mpoly_ctx_init_rand(ctx, state, 10, p);
nmod_mpoly_init(a, ctx);
nmod_mpoly_init(t, ctx);
n = FLINT_MAX(WORD(1), ctx->minfo->nvars);
nfacs = 5 + (8 + n_randint(state, 8))/n;
expbounds = FLINT_ARRAY_ALLOC(ctx->minfo->nvars, ulong);
for (k = 0; k < ctx->minfo->nvars; k++)
expbounds[k] = 3;
nmod_mpoly_one(a, ctx);
for (j = 0; j < nfacs; j++)
{
len = 1 + n_randint(state, 9);
if (ctx->minfo->nvars > 0)
{
k = n_randint(state, ctx->minfo->nvars);
expbounds[k] = 1;
}
nmod_mpoly_randtest_bounds(t, state, len, expbounds, ctx);
if (!nmod_mpoly_is_zero(t, ctx))
{
nmod_mpoly_mul(t, a, t, ctx);
if (t->length < 1500)
nmod_mpoly_swap(a, t, ctx);
}
}
check_content(a, ctx);
flint_free(expbounds);
nmod_mpoly_clear(t, ctx);
nmod_mpoly_clear(a, ctx);
nmod_mpoly_ctx_clear(ctx);
}
TEST_FUNCTION_END(state);
}