tests/test_I_isqrt.cc

   1 #include "test_I.h"
   2
   3 int test_I_isqrt (int iterations)
   4 {
   5         int error = 0;
   6         int i;
   7         // Check against "*".
   8         for (i = iterations; i > 0; i--) {
   9                 cl_I a = testrandom_I();
  10                 if (a >= 0) {
  11                         cl_I w;
  12                         bool squarep = isqrt(a,&w);
  13                         ASSERT1(w >= 0 && expt_pos(w,2) <= a && a < expt_pos(w+1,2), a);
  14                         ASSERT1(squarep ? w*w==a : w*w<a, a);
  15                 }
  16         }
  17         // Check certain special cases.
  18         for (i = iterations; i > 0; i--) {
  19                 cl_I a = abs(testrandom_I());
  20                 cl_I w;
  21                 // Check a^2 is a square.
  22                 ASSERT1(isqrt(a*a,&w) && w == a, a);
  23                 // Check a^2+1 is not a square, except when a=0.
  24                 ASSERT1((a==0) || (!isqrt(a*a+1,&w) && w == a), a);
  25                 // Check a^2+2*a is not a square, except when a=0.
  26                 ASSERT1(isqrt(a*(a+2),&w)==(a==0) && w == a, a);
  27         }
  28         return error;
  29 }