check/exam_pseries.cpp

   1 /** @File exam_pseries.cpp
   2  *
   3  *  Series expansion test (Laurent and Taylor series). */
   4
   5 /*
   6  *  GiNaC Copyright (C) 1999-2009 Johannes Gutenberg University Mainz, Germany
   7  *
   8  *  This program is free software; you can redistribute it and/or modify
   9  *  it under the terms of the GNU General Public License as published by
  10  *  the Free Software Foundation; either version 2 of the License, or
  11  *  (at your option) any later version.
  12  *
  13  *  This program is distributed in the hope that it will be useful,
  14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  *  GNU General Public License for more details.
  17  *
  18  *  You should have received a copy of the GNU General Public License
  19  *  along with this program; if not, write to the Free Software
  20  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  21  */
  22
  23 #include "ginac.h"
  24 using namespace GiNaC;
  25
  26 #include <iostream>
  27 using namespace std;
  28
  29 static symbol x("x");
  30
  31 static unsigned check_series(const ex &e, const ex &point, const ex &d, int order = 8)
  32 {
  33         ex es = e.series(x==point, order);
  34         ex ep = ex_to<pseries>(es).convert_to_poly();
  35         if (!(ep - d).expand().is_zero()) {
  36                 clog << "series expansion of " << e << " at " << point
  37                      << " erroneously returned " << ep << " (instead of " << d
  38                      << ")" << endl;
  39                 clog << tree << (ep-d) << dflt;
  40                 return 1;
  41         }
  42         return 0;
  43 }
  44
  45 // Series expansion
  46 static unsigned exam_series1()
  47 {
  48         using GiNaC::log;
  49
  50         symbol a("a");
  51         symbol b("b");
  52         unsigned result = 0;
  53         ex e, d;
  54
  55         e = pow(a+b, x);
  56         d = 1 + Order(pow(x, 1));
  57         result += check_series(e, 0, d, 1);
  58
  59         e = sin(x);
  60         d = x - pow(x, 3) / 6 + pow(x, 5) / 120 - pow(x, 7) / 5040 + Order(pow(x, 8));
  61         result += check_series(e, 0, d);
  62
  63         e = cos(x);
  64         d = 1 - pow(x, 2) / 2 + pow(x, 4) / 24 - pow(x, 6) / 720 + Order(pow(x, 8));
  65         result += check_series(e, 0, d);
  66
  67         e = exp(x);
  68         d = 1 + x + pow(x, 2) / 2 + pow(x, 3) / 6 + pow(x, 4) / 24 + pow(x, 5) / 120 + pow(x, 6) / 720 + pow(x, 7) / 5040 + Order(pow(x, 8));
  69         result += check_series(e, 0, d);
  70
  71         e = pow(1 - x, -1);
  72         d = 1 + x + pow(x, 2) + pow(x, 3) + pow(x, 4) + pow(x, 5) + pow(x, 6) + pow(x, 7) + Order(pow(x, 8));
  73         result += check_series(e, 0, d);
  74
  75         e = x + pow(x, -1);
  76         d = x + pow(x, -1);
  77         result += check_series(e, 0, d);
  78
  79         e = x + pow(x, -1);
  80         d = 2 + pow(x-1, 2) - pow(x-1, 3) + pow(x-1, 4) - pow(x-1, 5) + pow(x-1, 6) - pow(x-1, 7) + Order(pow(x-1, 8));
  81         result += check_series(e, 1, d);
  82
  83         e = pow(x + pow(x, 3), -1);
  84         d = pow(x, -1) - x + pow(x, 3) - pow(x, 5) + pow(x, 7) + Order(pow(x, 8));
  85         result += check_series(e, 0, d);
  86
  87         e = pow(pow(x, 2) + pow(x, 4), -1);
  88         d = pow(x, -2) - 1 + pow(x, 2) - pow(x, 4) + pow(x, 6) + Order(pow(x, 8));
  89         result += check_series(e, 0, d);
  90
  91         e = pow(sin(x), -2);
  92         d = pow(x, -2) + numeric(1,3) + pow(x, 2) / 15 + pow(x, 4) * 2/189 + pow(x, 6) / 675  + Order(pow(x, 8));
  93         result += check_series(e, 0, d);
  94
  95         e = sin(x) / cos(x);
  96         d = x + pow(x, 3) / 3 + pow(x, 5) * 2/15 + pow(x, 7) * 17/315 + Order(pow(x, 8));
  97         result += check_series(e, 0, d);
  98
  99         e = cos(x) / sin(x);
 100         d = pow(x, -1) - x / 3 - pow(x, 3) / 45 - pow(x, 5) * 2/945 - pow(x, 7) / 4725 + Order(pow(x, 8));
 101         result += check_series(e, 0, d);
 102
 103         e = pow(numeric(2), x);
 104         ex t = log(2) * x;
 105         d = 1 + t + pow(t, 2) / 2 + pow(t, 3) / 6 + pow(t, 4) / 24 + pow(t, 5) / 120 + pow(t, 6) / 720 + pow(t, 7) / 5040 + Order(pow(x, 8));
 106         result += check_series(e, 0, d.expand());
 107
 108         e = pow(Pi, x);
 109         t = log(Pi) * x;
 110         d = 1 + t + pow(t, 2) / 2 + pow(t, 3) / 6 + pow(t, 4) / 24 + pow(t, 5) / 120 + pow(t, 6) / 720 + pow(t, 7) / 5040 + Order(pow(x, 8));
 111         result += check_series(e, 0, d.expand());
 112
 113         e = log(x);
 114         d = e;
 115         result += check_series(e, 0, d, 1);
 116         result += check_series(e, 0, d, 2);
 117
 118         e = pow(x, 8) * pow(pow(x,3)+ pow(x + pow(x,3), 2), -2);
 119         d = pow(x, 4) - 2*pow(x, 5) + Order(pow(x, 6));
 120         result += check_series(e, 0, d, 6);
 121
 122         e = cos(x) * pow(sin(x)*(pow(x, 5) + 4 * pow(x, 2)), -3);
 123         d = pow(x, -9) / 64 - 3 * pow(x, -6) / 256 - pow(x, -5) / 960 + 535 * pow(x, -3) / 96768
 124             + pow(x, -2) / 1280 - pow(x, -1) / 14400 - numeric(283, 129024) - 2143 * x / 5322240
 125             + Order(pow(x, 2));
 126         result += check_series(e, 0, d, 2);
 127
 128         e = sqrt(1+x*x) * sqrt(1+2*x*x);
 129         d = 1 + Order(pow(x, 2));
 130         result += check_series(e, 0, d, 2);
 131
 132         e = pow(x, 4) * sin(a) + pow(x, 2);
 133         d = pow(x, 2) + Order(pow(x, 3));
 134         result += check_series(e, 0, d, 3);
 135
 136         e = log(a*x + b*x*x*log(x));
 137         d = log(a*x) + b/a*log(x)*x - pow(b/a, 2)/2*pow(log(x)*x, 2) + Order(pow(x, 3));
 138         result += check_series(e, 0, d, 3);
 139
 140         e = pow((x+a), b);
 141         d = pow(a, b) + (pow(a, b)*b/a)*x + (pow(a, b)*b*b/a/a/2 - pow(a, b)*b/a/a/2)*pow(x, 2) + Order(pow(x, 3));
 142         result += check_series(e, 0, d, 3);
 143
 144         return result;
 145 }
 146
 147 // Series addition
 148 static unsigned exam_series2()
 149 {
 150         unsigned result = 0;
 151         ex e, d;
 152
 153         e = pow(sin(x), -1).series(x==0, 8) + pow(sin(-x), -1).series(x==0, 12);
 154         d = Order(pow(x, 8));
 155         result += check_series(e, 0, d);
 156
 157         return result;
 158 }
 159
 160 // Series multiplication
 161 static unsigned exam_series3()
 162 {
 163         unsigned result = 0;
 164         ex e, d;
 165
 166         e = sin(x).series(x==0, 8) * pow(sin(x), -1).series(x==0, 12);
 167         d = 1 + Order(pow(x, 7));
 168         result += check_series(e, 0, d);
 169
 170         return result;
 171 }
 172
 173 // Series exponentiation
 174 static unsigned exam_series4()
 175 {
 176         unsigned result = 0;
 177         ex e, d;
 178
 179         e = pow((2*cos(x)).series(x==0, 5), 2).series(x==0, 5);
 180         d = 4 - 4*pow(x, 2) + 4*pow(x, 4)/3 + Order(pow(x, 5));
 181         result += check_series(e, 0, d);
 182
 183         e = pow(tgamma(x), 2).series(x==0, 2);
 184         d = pow(x,-2) - 2*Euler/x + (pow(Pi,2)/6+2*pow(Euler,2))
 185                 + x*(-4*pow(Euler, 3)/3 -pow(Pi,2)*Euler/3 - 2*zeta(3)/3) + Order(pow(x, 2));
 186         result += check_series(e, 0, d);
 187
 188         return result;
 189 }
 190
 191 // Order term handling
 192 static unsigned exam_series5()
 193 {
 194         unsigned result = 0;
 195         ex e, d;
 196
 197         e = 1 + x + pow(x, 2) + pow(x, 3);
 198         d = Order(1);
 199         result += check_series(e, 0, d, 0);
 200         d = 1 + Order(x);
 201         result += check_series(e, 0, d, 1);
 202         d = 1 + x + Order(pow(x, 2));
 203         result += check_series(e, 0, d, 2);
 204         d = 1 + x + pow(x, 2) + Order(pow(x, 3));
 205         result += check_series(e, 0, d, 3);
 206         d = 1 + x + pow(x, 2) + pow(x, 3);
 207         result += check_series(e, 0, d, 4);
 208         return result;
 209 }
 210
 211 // Series expansion of tgamma(-1)
 212 static unsigned exam_series6()
 213 {
 214         ex e = tgamma(2*x);
 215         ex d = pow(x+1,-1)*numeric(1,4) +
 216                pow(x+1,0)*(numeric(3,4) -
 217                            numeric(1,2)*Euler) +
 218                pow(x+1,1)*(numeric(7,4) -
 219                            numeric(3,2)*Euler +
 220                            numeric(1,2)*pow(Euler,2) +
 221                            numeric(1,12)*pow(Pi,2)) +
 222                pow(x+1,2)*(numeric(15,4) -
 223                            numeric(7,2)*Euler -
 224                            numeric(1,3)*pow(Euler,3) +
 225                            numeric(1,4)*pow(Pi,2) +
 226                            numeric(3,2)*pow(Euler,2) -
 227                            numeric(1,6)*pow(Pi,2)*Euler -
 228                            numeric(2,3)*zeta(3)) +
 229                pow(x+1,3)*(numeric(31,4) - pow(Euler,3) -
 230                            numeric(15,2)*Euler +
 231                            numeric(1,6)*pow(Euler,4) +
 232                            numeric(7,2)*pow(Euler,2) +
 233                            numeric(7,12)*pow(Pi,2) -
 234                            numeric(1,2)*pow(Pi,2)*Euler -
 235                            numeric(2)*zeta(3) +
 236                            numeric(1,6)*pow(Euler,2)*pow(Pi,2) +
 237                            numeric(1,40)*pow(Pi,4) +
 238                            numeric(4,3)*zeta(3)*Euler) +
 239                Order(pow(x+1,4));
 240         return check_series(e, -1, d, 4);
 241 }
 242
 243 // Series expansion of tan(x==Pi/2)
 244 static unsigned exam_series7()
 245 {
 246         ex e = tan(x*Pi/2);
 247         ex d = pow(x-1,-1)/Pi*(-2) + pow(x-1,1)*Pi/6 + pow(x-1,3)*pow(Pi,3)/360
 248               +pow(x-1,5)*pow(Pi,5)/15120 + pow(x-1,7)*pow(Pi,7)/604800
 249               +Order(pow(x-1,9));
 250         return check_series(e,1,d,9);
 251 }
 252
 253 // Series expansion of log(sin(x==0))
 254 static unsigned exam_series8()
 255 {
 256         ex e = log(sin(x));
 257         ex d = log(x) - pow(x,2)/6 - pow(x,4)/180 - pow(x,6)/2835 - pow(x,8)/37800 + Order(pow(x,9));
 258         return check_series(e,0,d,9);
 259 }
 260
 261 // Series expansion of Li2(sin(x==0))
 262 static unsigned exam_series9()
 263 {
 264         ex e = Li2(sin(x));
 265         ex d = x + pow(x,2)/4 - pow(x,3)/18 - pow(x,4)/48
 266                - 13*pow(x,5)/1800 - pow(x,6)/360 - 23*pow(x,7)/21168
 267                + Order(pow(x,8));
 268         return check_series(e,0,d,8);
 269 }
 270
 271 // Series expansion of Li2((x==2)^2), caring about branch-cut
 272 static unsigned exam_series10()
 273 {
 274         using GiNaC::log;
 275
 276         ex e = Li2(pow(x,2));
 277         ex d = Li2(4) + (-log(3) + I*Pi*csgn(I-I*pow(x,2))) * (x-2)
 278                + (numeric(-2,3) + log(3)/4 - I*Pi/4*csgn(I-I*pow(x,2))) * pow(x-2,2)
 279                + (numeric(11,27) - log(3)/12 + I*Pi/12*csgn(I-I*pow(x,2))) * pow(x-2,3)
 280                + (numeric(-155,648) + log(3)/32 - I*Pi/32*csgn(I-I*pow(x,2))) * pow(x-2,4)
 281                + Order(pow(x-2,5));
 282         return check_series(e,2,d,5);
 283 }
 284
 285 // Series expansion of logarithms around branch points
 286 static unsigned exam_series11()
 287 {
 288         using GiNaC::log;
 289
 290         unsigned result = 0;
 291         ex e, d;
 292         symbol a("a");
 293
 294         e = log(x);
 295         d = log(x);
 296         result += check_series(e,0,d,5);
 297
 298         e = log(3/x);
 299         d = log(3)-log(x);
 300         result += check_series(e,0,d,5);
 301
 302         e = log(3*pow(x,2));
 303         d = log(3)+2*log(x);
 304         result += check_series(e,0,d,5);
 305
 306         // These ones must not be expanded because it would result in a branch cut
 307         // running in the wrong direction. (Other systems tend to get this wrong.)
 308         e = log(-x);
 309         d = e;
 310         result += check_series(e,0,d,5);
 311
 312         e = log(I*(x-123));
 313         d = e;
 314         result += check_series(e,123,d,5);
 315
 316         e = log(a*x);
 317         d = e;  // we don't know anything about a!
 318         result += check_series(e,0,d,5);
 319
 320         e = log((1-x)/x);
 321         d = log(1-x) - (x-1) + pow(x-1,2)/2 - pow(x-1,3)/3  + pow(x-1,4)/4 + Order(pow(x-1,5));
 322         result += check_series(e,1,d,5);
 323
 324         return result;
 325 }
 326
 327 // Series expansion of other functions around branch points
 328 static unsigned exam_series12()
 329 {
 330         using GiNaC::log;
 331
 332         unsigned result = 0;
 333         ex e, d;
 334
 335         // NB: Mma and Maple give different results, but they agree if one
 336         // takes into account that by assumption |x|<1.
 337         e = atan(x);
 338         d = (I*log(2)/2-I*log(1+I*x)/2) + (x-I)/4 + I*pow(x-I,2)/16 + Order(pow(x-I,3));
 339         result += check_series(e,I,d,3);
 340
 341         // NB: here, at -I, Mathematica disagrees, but it is wrong -- they
 342         // pick up a complex phase by incorrectly expanding logarithms.
 343         e = atan(x);
 344         d = (-I*log(2)/2+I*log(1-I*x)/2) + (x+I)/4 - I*pow(x+I,2)/16 + Order(pow(x+I,3));
 345         result += check_series(e,-I,d,3);
 346
 347         // This is basically the same as above, the branch point is at +/-1:
 348         e = atanh(x);
 349         d = (-log(2)/2+log(x+1)/2) + (x+1)/4 + pow(x+1,2)/16 + Order(pow(x+1,3));
 350         result += check_series(e,-1,d,3);
 351
 352         return result;
 353 }
 354
 355 // Test of the patch of Stefan Weinzierl that prevents an infinite loop if
 356 // a factor in a product is a complicated way of writing zero.
 357 static unsigned exam_series13()
 358 {
 359         unsigned result = 0;
 360
 361         ex e = (new mul(pow(2,x), (1/x*(-(1+x)/(1-x)) + (1+x)/x/(1-x)))
 362                )->setflag(status_flags::evaluated);
 363         ex d = Order(x);
 364         result += check_series(e,0,d,1);
 365
 366         return result;
 367 }
 368
 369 unsigned exam_pseries()
 370 {
 371         unsigned result = 0;
 372
 373         cout << "examining series expansion" << flush;
 374
 375         result += exam_series1();  cout << '.' << flush;
 376         result += exam_series2();  cout << '.' << flush;
 377         result += exam_series3();  cout << '.' << flush;
 378         result += exam_series4();  cout << '.' << flush;
 379         result += exam_series5();  cout << '.' << flush;
 380         result += exam_series6();  cout << '.' << flush;
 381         result += exam_series7();  cout << '.' << flush;
 382         result += exam_series8();  cout << '.' << flush;
 383         result += exam_series9();  cout << '.' << flush;
 384         result += exam_series10();  cout << '.' << flush;
 385         result += exam_series11();  cout << '.' << flush;
 386         result += exam_series12();  cout << '.' << flush;
 387         result += exam_series13();  cout << '.' << flush;
 388
 389         return result;
 390 }
 391
 392 int main(int argc, char** argv)
 393 {
 394         return exam_pseries();
 395 }