contrib/libgmp/mpz/tests/t-sqrtrem.c

   1 /* Test mpz_add, mpz_add_ui, mpz_cmp, mpz_cmp, mpz_mul, mpz_sqrtrem.
   2
   3 Copyright (C) 1991, 1993, 1994, 1996 Free Software Foundation, Inc.
   4
   5 This file is part of the GNU MP Library.
   6
   7 The GNU MP Library is free software; you can redistribute it and/or modify
   8 it under the terms of the GNU Library General Public License as published by
   9 the Free Software Foundation; either version 2 of the License, or (at your
  10 option) any later version.
  11
  12 The GNU MP Library is distributed in the hope that it will be useful, but
  13 WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  14 or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
  15 License for more details.
  16
  17 You should have received a copy of the GNU Library General Public License
  18 along with the GNU MP Library; see the file COPYING.LIB.  If not, write to
  19 the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  20 MA 02111-1307, USA. */
  21
  22 #include <stdio.h>
  23 #include "gmp.h"
  24 #include "gmp-impl.h"
  25 #include "urandom.h"
  26
  27 void debug_mp ();
  28
  29 #ifndef SIZE
  30 #define SIZE 16
  31 #endif
  32
  33 main (argc, argv)
  34      int argc;
  35      char **argv;
  36 {
  37   mpz_t x2;
  38   mpz_t x, rem;
  39   mpz_t temp, temp2;
  40   mp_size_t x2_size;
  41   int i;
  42   int reps = 100000;
  43
  44   if (argc == 2)
  45      reps = atoi (argv[1]);
  46
  47   mpz_init (x2);
  48   mpz_init (x);
  49   mpz_init (rem);
  50   mpz_init (temp);
  51   mpz_init (temp2);
  52
  53   for (i = 0; i < reps; i++)
  54     {
  55       x2_size = urandom () % SIZE;
  56
  57       mpz_random2 (x2, x2_size);
  58
  59       mpz_sqrtrem (x, rem, x2);
  60       mpz_mul (temp, x, x);
  61
  62       /* Is square of result > argument?  */
  63       if (mpz_cmp (temp, x2) > 0)
  64         dump_abort (x2, x, rem);
  65
  66       mpz_add_ui (temp2, x, 1);
  67       mpz_mul (temp2, temp2, temp2);
  68
  69       /* Is square of (result + 1) <= argument?  */
  70       if (mpz_cmp (temp2, x2) <= 0)
  71         dump_abort (x2, x, rem);
  72
  73       mpz_add (temp2, temp, rem);
  74
  75       /* Is the remainder wrong?  */
  76       if (mpz_cmp (x2, temp2) != 0)
  77         dump_abort (x2, x, rem);
  78     }
  79
  80   exit (0);
  81 }
  82
  83 dump_abort (x2, x, rem)
  84      mpz_t x2, x, rem;
  85 {
  86   fprintf (stderr, "ERROR\n");
  87   fprintf (stderr, "x2        = "); debug_mp (x2, -16);
  88   fprintf (stderr, "x         = "); debug_mp (x, -16);
  89   fprintf (stderr, "remainder = "); debug_mp (rem, -16);
  90   abort();
  91 }
  92
  93 void
  94 debug_mp (x, base)
  95      mpz_t x;
  96 {
  97   mpz_out_str (stderr, base, x); fputc ('\n', stderr);
  98 }