mpir/tests/mpn/t-sizeinbase.c

109 lines
2.3 KiB
C

/* Test for sizeinbase function.
Copyright 2014 Free Software Foundation, Inc.
This file is part of the GNU MP Library test suite.
The GNU MP Library test suite is free software; you can redistribute it
and/or modify it under the terms of the GNU General Public License as
published by the Free Software Foundation; either version 3 of the License,
or (at your option) any later version.
The GNU MP Library test suite is distributed in the hope that it will be
useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
Public License for more details.
You should have received a copy of the GNU General Public License along with
the GNU MP Library test suite. If not, see https://www.gnu.org/licenses/. */
#include <stdlib.h>
#include <stdio.h>
#include "mpir.h"
#include "gmp-impl.h"
#include "tests.h"
/* Exponents up to 2^SIZE_LOG */
#ifndef SIZE_LOG
#define SIZE_LOG 13
#endif
#ifndef COUNT
#define COUNT 30
#endif
#define MAX_N (1<<SIZE_LOG)
int
main (int argc, char **argv)
{
mp_limb_t a;
mp_ptr pp, scratch;
mp_limb_t max_b;
int count = COUNT;
int test;
gmp_randstate_ptr rands;
TMP_DECL;
if (argc > 1)
{
char *end;
count = strtol (argv[1], &end, 0);
if (*end || count <= 0)
{
fprintf (stderr, "Invalid test count: %s.\n", argv[1]);
return 1;
}
}
tests_start ();
TMP_MARK;
rands = RANDS;
pp = TMP_ALLOC_LIMBS (MAX_N);
scratch = TMP_ALLOC_LIMBS (MAX_N);
max_b = numberof (mp_bases);
ASSERT_ALWAYS (max_b > 62);
ASSERT_ALWAYS (max_b < GMP_NUMB_MAX);
for (a = 2; a < max_b; ++a)
for (test = 0; test < count; ++test)
{
mp_size_t pn;
mp_limb_t exp;
mp_bitcnt_t res;
exp = gmp_urandomm_ui (rands, MAX_N);
pn = mpn_pow_1 (pp, &a, 1, exp, scratch);
res = mpn_sizeinbase (pp, pn, a) - 1;
if ((res < exp) || (res > exp + 1))
{
printf ("ERROR in test %d, base = %d, exp = %d, res = %d\n",
test, (int) a, (int) exp, (int) res);
abort();
}
mpn_sub_1 (pp, pp, pn, CNST_LIMB(1));
pn -= pp[pn-1] == 0;
res = mpn_sizeinbase (pp, pn, a);
if ((res < exp) || (res - 1 > exp))
{
printf ("ERROR in -1 test %d, base = %d, exp = %d, res = %d\n",
test, (int) a, (int) exp, (int) res);
abort();
}
}
TMP_FREE;
tests_end ();
return 0;
}