added mpfr_root

git-svn-id: svn://scm.gforge.inria.fr/svn/mpfr/trunk@3475 280ebfd0-de03-0410-8827-d642c229c3f4

1 files changed, 224 insertions, 0 deletions

diff --git a/tests/troot.c b/tests/troot.c
new file mode 100644
index 000000000..525f74370
--- /dev/null
+++ b/tests/troot.c
@@ -0,0 +1,224 @@
+/* Test file for mpfr_root.
+
+Copyright 2005 Free Software Foundation, Inc.
+
+This file is part of the MPFR Library.
+
+The MPFR Library is free software; you can redistribute it and/or modify
+it under the terms of the GNU Lesser General Public License as published by
+the Free Software Foundation; either version 2.1 of the License, or (at your
+option) any later version.
+
+The MPFR Library 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 Lesser General Public
+License for more details.
+
+You should have received a copy of the GNU Lesser General Public License
+along with the MPFR Library; see the file COPYING.LIB.  If not, write to
+the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
+MA 02111-1307, USA. */
+
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "mpfr-test.h"
+
+static void
+special (void)
+{
+  mpfr_t x, y;
+
+  mpfr_init (x);
+  mpfr_init (y);
+
+  /* root(NaN) = NaN */
+  mpfr_set_nan (x);
+  mpfr_root (y, x, 17, GMP_RNDN);
+  if (!mpfr_nan_p (y))
+    {
+      printf ("Error: root(NaN,17) <> NaN\n");
+      exit (1);
+    }
+
+  /* root(+Inf) = +Inf */
+  mpfr_set_inf (x, 1);
+  mpfr_root (y, x, 42, GMP_RNDN);
+  if (!mpfr_inf_p (y) || mpfr_sgn (y) < 0)
+    {
+      printf ("Error: root(+Inf,42) <> +Inf\n");
+      exit (1);
+    }
+
+  /* root(-Inf, 17) =  -Inf */
+  mpfr_set_inf (x, -1);
+  mpfr_root (y, x, 17, GMP_RNDN);
+  if (!mpfr_inf_p (y) || mpfr_sgn (y) > 0)
+    {
+      printf ("Error: root(-Inf,17) <> -Inf\n");
+      exit (1);
+    }
+  /* root(-Inf, 42) =  NaN */
+  mpfr_set_inf (x, -1);
+  mpfr_root (y, x, 42, GMP_RNDN);
+  if (!mpfr_nan_p (y))
+    {
+      printf ("Error: root(-Inf,42) <> -Inf\n");
+      exit (1);
+    }
+
+  /* root(+/-0) =  +/-0 */
+  mpfr_set_ui (x, 0, GMP_RNDN);
+  mpfr_root (y, x, 17, GMP_RNDN);
+  if (mpfr_cmp_ui (y, 0) || mpfr_sgn (y) < 0)
+    {
+      printf ("Error: root(+0,17) <> +0\n");
+      exit (1);
+    }
+  mpfr_neg (x, x, GMP_RNDN);
+  mpfr_root (y, x, 42, GMP_RNDN);
+  if (mpfr_cmp_ui (y, 0) || mpfr_sgn (y) > 0)
+    {
+      printf ("Error: root(-0,42) <> -0\n");
+      exit (1);
+    }
+
+  mpfr_set_prec (x, 53);
+  mpfr_set_str (x, "8.39005285514734966412e-01", 10, GMP_RNDN);
+  mpfr_root (x, x, 3, GMP_RNDN);
+  if (mpfr_cmp_str1 (x, "9.43166207799662426048e-01"))
+    {
+      printf ("Error in root3 (1)\n");
+      printf ("expected 9.43166207799662426048e-01\n");
+      printf ("got      ");
+      mpfr_dump (x);
+      exit (1);
+    }
+
+  mpfr_set_prec (x, 32);
+  mpfr_set_prec (y, 32);
+  mpfr_set_str_binary (x, "0.10000100001100101001001001011001");
+  mpfr_root (x, x, 3, GMP_RNDN);
+  mpfr_set_str_binary (y, "0.11001101011000100111000111111001");
+  if (mpfr_cmp (x, y))
+    {
+      printf ("Error in root3 (2)\n");
+      exit (1);
+    }
+
+  mpfr_set_prec (x, 32);
+  mpfr_set_prec (y, 32);
+  mpfr_set_str_binary (x, "-0.1100001110110000010101011001011");
+  mpfr_root (x, x, 3, GMP_RNDD);
+  mpfr_set_str_binary (y, "-0.11101010000100100101000101011001");
+  if (mpfr_cmp (x, y))
+    {
+      printf ("Error in root3 (3)\n");
+      exit (1);
+    }
+
+  mpfr_set_prec (x, 82);
+  mpfr_set_prec (y, 27);
+  mpfr_set_str_binary (x, "0.1010001111011101011011000111001011001101100011110110010011011011011010011001100101e-7");
+  mpfr_root (y, x, 3, GMP_RNDD);
+  mpfr_set_str_binary (x, "0.101011110001110001000100011E-2");
+  if (mpfr_cmp (x, y))
+    {
+      printf ("Error in root3 (4)\n");
+      exit (1);
+    }
+
+  mpfr_set_prec (x, 204);
+  mpfr_set_prec (y, 38);
+  mpfr_set_str_binary (x, "0.101000000001101000000001100111111011111001110110100001111000100110100111001101100111110001110001011011010110010011100101111001111100001010010100111011101100000011011000101100010000000011000101001010001001E-5");
+  mpfr_root (y, x, 3, GMP_RNDD);
+  mpfr_set_str_binary (x, "0.10001001111010011011101000010110110010E-1");
+  if (mpfr_cmp (x, y))
+    {
+      printf ("Error in root3 (5)\n");
+      exit (1);
+    }
+
+  mpfr_clear (x);
+  mpfr_clear (y);
+}
+
+#define TEST_FUNCTION mpfr_root
+#define TWO_ARGS_UI
+#include "tgeneric.c"
+
+int
+main (void)
+{
+  mpfr_t x;
+  int r;
+  mp_prec_t p;
+  unsigned long k;
+
+  MPFR_TEST_USE_RANDS ();
+  tests_start_mpfr ();
+
+  special ();
+
+  mpfr_init (x);
+
+  for (p = 2; p < 100; p++)
+    {
+      mpfr_set_prec (x, p);
+      for (r = 0; r < GMP_RND_MAX; r++)
+        {
+          mpfr_set_ui (x, 1, GMP_RNDN);
+          k = 2 + randlimb () % 4; /* 2 <= k <= 5 */
+          mpfr_root (x, x, k, (mp_rnd_t) r);
+          if (mpfr_cmp_ui (x, 1))
+            {
+              printf ("Error in mpfr_root(%lu) for x=1, rnd=%s\ngot ",
+                      k, mpfr_print_rnd_mode ((mp_rnd_t) r));
+              mpfr_out_str (stdout, 2, 0, x, GMP_RNDN);
+              printf ("\n");
+              exit (1);
+            }
+          mpfr_set_si (x, -1, GMP_RNDN);
+          if (k % 2)
+            {
+              mpfr_root (x, x, k, (mp_rnd_t) r);
+              if (mpfr_cmp_si (x, -1))
+                {
+                  printf ("Error in mpfr_root(%lu) for x=-1, rnd=%s\ngot ",
+                          k, mpfr_print_rnd_mode ((mp_rnd_t) r));
+                  mpfr_out_str (stdout, 2, 0, x, GMP_RNDN);
+                  printf ("\n");
+                  exit (1);
+                }
+            }
+
+          if (p >= 5)
+            {
+              int i;
+              for (i = -12; i <= 12; i++)
+                {
+                  mpfr_set_ui (x, 27, GMP_RNDN);
+                  mpfr_mul_2si (x, x, 3*i, GMP_RNDN);
+                  mpfr_root (x, x, 3, GMP_RNDN);
+                  if (mpfr_cmp_si_2exp (x, 3, i))
+                    {
+                      printf ("Error in mpfr_root(3) for "
+                              "x = 27.0 * 2^(%d), rnd=%s\ngot ",
+                              3*i, mpfr_print_rnd_mode ((mp_rnd_t) r));
+                      mpfr_out_str (stdout, 2, 0, x, GMP_RNDN);
+                      printf ("\ninstead of 3 * 2^(%d)\n", i);
+                      exit (1);
+                    }
+                }
+            }
+        }
+    }
+  mpfr_clear (x);
+
+  test_generic (2, 200, 3, 10);
+  test_generic (2, 200, 4, 10);
+  test_generic (2, 200, 5, 10);
+
+  tests_end_mpfr ();
+  return 0;
+}