#include "test_helpers.h"
#include "fmpz_mpoly.h"
TEST_FUNCTION_START(fmpz_mpoly_get_set_is_fmpz, state)
{
slong i;
for (i = 0; i < 1000 * flint_test_multiplier(); i++)
{
fmpz_mpoly_ctx_t ctx;
fmpz_mpoly_t f;
fmpz_t c, d;
slong len;
flint_bitcnt_t coeff_bits, exp_bits;
fmpz_init(c);
fmpz_init(d);
fmpz_mpoly_ctx_init_rand(ctx, state, 20);
fmpz_mpoly_init(f, ctx);
len = n_randint(state, 10);
exp_bits = n_randint(state, 200);
coeff_bits = n_randint(state, 200);
fmpz_mpoly_randtest_bits(f, state, len, coeff_bits, exp_bits, ctx);
if (fmpz_mpoly_is_fmpz(f, ctx))
{
fmpz_mpoly_get_fmpz(c, f, ctx);
if (!fmpz_mpoly_equal_fmpz(f, c, ctx))
{
printf("FAIL\n");
flint_printf("Check is_fmpz and get_fmpz catch constants\ni = %wd\n", i);
fflush(stdout);
flint_abort();
}
}
fmpz_randtest(c, state, n_randint(state, 200));
fmpz_mpoly_set_fmpz(f, c, ctx);
if (!fmpz_mpoly_is_fmpz(f, ctx))
{
printf("FAIL\n");
flint_printf("Check set_fmpz makes is_fmpz true\ni = %wd\n", i);
fflush(stdout);
flint_abort();
}
fmpz_mpoly_get_fmpz(d, f, ctx);
if (!fmpz_equal(c, d))
{
printf("FAIL\n");
flint_printf("Check get_fmpz matches set_fmpz true\ni = %wd\n", i);
fflush(stdout);
flint_abort();
}
fmpz_clear(c);
fmpz_clear(d);
fmpz_mpoly_clear(f, ctx);
fmpz_mpoly_ctx_clear(ctx);
}
TEST_FUNCTION_END(state);
}