1 /* Copyright 2012 Free Software Foundation, Inc. 2 3 This file is part of the GNU MP Library test suite. 4 5 The GNU MP Library test suite is free software; you can redistribute it 6 and/or modify it under the terms of the GNU General Public License as 7 published by the Free Software Foundation; either version 3 of the License, 8 or (at your option) any later version. 9 10 The GNU MP Library test suite is distributed in the hope that it will be 11 useful, but WITHOUT ANY WARRANTY; without even the implied warranty of 12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General 13 Public License for more details. 14 15 You should have received a copy of the GNU General Public License along with 16 the GNU MP Library test suite. If not, see http://www.gnu.org/licenses/. */ 17 18 19 #include <stdlib.h> /* for strtol */ 20 #include <stdio.h> /* for printf */ 21 22 #include "gmp.h" 23 #include "gmp-impl.h" 24 #include "longlong.h" 25 #include "tests/tests.h" 26 27 #define MAX_LIMBS 150 28 #define COUNT 500 29 30 int 31 main (int argc, char **argv) 32 { 33 gmp_randstate_ptr rands; 34 35 mp_ptr ap, rp, pp, scratch; 36 int count = COUNT; 37 unsigned i; 38 TMP_DECL; 39 40 TMP_MARK; 41 42 if (argc > 1) 43 { 44 char *end; 45 count = strtol (argv[1], &end, 0); 46 if (*end || count <= 0) 47 { 48 fprintf (stderr, "Invalid test count: %s.\n", argv[1]); 49 return 1; 50 } 51 } 52 53 tests_start (); 54 rands = RANDS; 55 56 ap = TMP_ALLOC_LIMBS (MAX_LIMBS); 57 rp = TMP_ALLOC_LIMBS (MAX_LIMBS); 58 pp = TMP_ALLOC_LIMBS (MAX_LIMBS); 59 scratch = TMP_ALLOC_LIMBS (3*MAX_LIMBS); /* For mpn_powlo */ 60 61 for (i = 0; i < count; i++) 62 { 63 mp_size_t n; 64 mp_limb_t k; 65 int c; 66 67 n = 1 + gmp_urandomm_ui (rands, MAX_LIMBS); 68 69 if (i & 1) 70 mpn_random2 (ap, n); 71 else 72 mpn_random (ap, n); 73 74 ap[0] |= 1; 75 76 if (i < 100) 77 k = 3 + 2*i; 78 else 79 { 80 mpn_random (&k, 1); 81 if (k < 3) 82 k = 3; 83 else 84 k |= 1; 85 } 86 mpn_broot (rp, ap, n, k); 87 mpn_powlo (pp, rp, &k, 1, n, scratch); 88 89 MPN_CMP (c, ap, pp, n); 90 if (c != 0) 91 { 92 gmp_fprintf (stderr, 93 "mpn_broot returned bad result: %u limbs\n", 94 (unsigned) n); 95 gmp_fprintf (stderr, "k = %Mx\n", k); 96 gmp_fprintf (stderr, "a = %Nx\n", ap, n); 97 gmp_fprintf (stderr, "r = %Nx\n", rp, n); 98 gmp_fprintf (stderr, "r^n = %Nx\n", pp, n); 99 abort (); 100 } 101 } 102 TMP_FREE; 103 tests_end (); 104 return 0; 105 } 106