Synced to HEAD.
[ginac.git] / 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-2004 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
21  */
22
23 #include "exams.h"
24
25 static symbol x("x");
26
27 static unsigned check_series(const ex &e, const ex &point, const ex &d, int order = 8)
28 {
29         ex es = e.series(x==point, order);
30         ex ep = ex_to<pseries>(es).convert_to_poly();
31         if (!(ep - d).expand().is_zero()) {
32                 clog << "series expansion of " << e << " at " << point
33                      << " erroneously returned " << ep << " (instead of " << d
34                      << ")" << endl;
35                 clog << tree << (ep-d) << dflt;
36                 return 1;
37         }
38         return 0;
39 }
40
41 // Series expansion
42 static unsigned exam_series1()
43 {
44         using GiNaC::log;
45
46         unsigned result = 0;
47         ex e, d;
48         
49         e = sin(x);
50         d = x - pow(x, 3) / 6 + pow(x, 5) / 120 - pow(x, 7) / 5040 + Order(pow(x, 8));
51         result += check_series(e, 0, d);
52         
53         e = cos(x);
54         d = 1 - pow(x, 2) / 2 + pow(x, 4) / 24 - pow(x, 6) / 720 + Order(pow(x, 8));
55         result += check_series(e, 0, d);
56         
57         e = exp(x);
58         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));
59         result += check_series(e, 0, d);
60         
61         e = pow(1 - x, -1);
62         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));
63         result += check_series(e, 0, d);
64         
65         e = x + pow(x, -1);
66         d = x + pow(x, -1);
67         result += check_series(e, 0, d);
68         
69         e = x + pow(x, -1);
70         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));
71         result += check_series(e, 1, d);
72         
73         e = pow(x + pow(x, 3), -1);
74         d = pow(x, -1) - x + pow(x, 3) - pow(x, 5) + pow(x, 7) + Order(pow(x, 8));
75         result += check_series(e, 0, d);
76         
77         e = pow(pow(x, 2) + pow(x, 4), -1);
78         d = pow(x, -2) - 1 + pow(x, 2) - pow(x, 4) + pow(x, 6) + Order(pow(x, 8));
79         result += check_series(e, 0, d);
80         
81         e = pow(sin(x), -2);
82         d = pow(x, -2) + numeric(1,3) + pow(x, 2) / 15 + pow(x, 4) * 2/189 + pow(x, 6) / 675  + Order(pow(x, 8));
83         result += check_series(e, 0, d);
84         
85         e = sin(x) / cos(x);
86         d = x + pow(x, 3) / 3 + pow(x, 5) * 2/15 + pow(x, 7) * 17/315 + Order(pow(x, 8));
87         result += check_series(e, 0, d);
88         
89         e = cos(x) / sin(x);
90         d = pow(x, -1) - x / 3 - pow(x, 3) / 45 - pow(x, 5) * 2/945 - pow(x, 7) / 4725 + Order(pow(x, 8));
91         result += check_series(e, 0, d);
92         
93         e = pow(numeric(2), x);
94         ex t = log(2) * x;
95         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));
96         result += check_series(e, 0, d.expand());
97         
98         e = pow(Pi, x);
99         t = log(Pi) * x;
100         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));
101         result += check_series(e, 0, d.expand());
102         
103         e = log(x);
104         d = e;
105         result += check_series(e, 0, d, 1);
106         result += check_series(e, 0, d, 2);
107         
108         e = pow(x, 8) * pow(pow(x,3)+ pow(x + pow(x,3), 2), -2);
109         d = pow(x, 4) - 2*pow(x, 5) + Order(pow(x, 6));
110         result += check_series(e, 0, d, 6);
111         
112         e = cos(x) * pow(sin(x)*(pow(x, 5) + 4 * pow(x, 2)), -3);
113         d = pow(x, -9) / 64 - 3 * pow(x, -6) / 256 - pow(x, -5) / 960 + 535 * pow(x, -3) / 96768
114             + pow(x, -2) / 1280 - pow(x, -1) / 14400 - numeric(283, 129024) - 2143 * x / 5322240
115             + Order(pow(x, 2));
116         result += check_series(e, 0, d, 2);
117         
118         symbol a("a");
119         e = pow(x, 4) * sin(a) + pow(x, 2);
120         d = pow(x, 2) + Order(pow(x, 3));
121         result += check_series(e, 0, d, 3);
122
123         return result;
124 }
125
126 // Series addition
127 static unsigned exam_series2()
128 {
129         unsigned result = 0;
130         ex e, d;
131         
132         e = pow(sin(x), -1).series(x==0, 8) + pow(sin(-x), -1).series(x==0, 12);
133         d = Order(pow(x, 8));
134         result += check_series(e, 0, d);
135         
136         return result;
137 }
138
139 // Series multiplication
140 static unsigned exam_series3()
141 {
142         unsigned result = 0;
143         ex e, d;
144         
145         e = sin(x).series(x==0, 8) * pow(sin(x), -1).series(x==0, 12);
146         d = 1 + Order(pow(x, 7));
147         result += check_series(e, 0, d);
148         
149         return result;
150 }
151
152 // Series exponentiation
153 static unsigned exam_series4()
154 {
155         unsigned result = 0;
156         ex e, d;
157         
158         e = pow((2*cos(x)).series(x==0, 5), 2).series(x==0, 5);
159         d = 4 - 4*pow(x, 2) + 4*pow(x, 4)/3 + Order(pow(x, 5));
160         result += check_series(e, 0, d);
161         
162         e = pow(tgamma(x), 2).series(x==0, 2);
163         d = pow(x,-2) - 2*Euler/x + (pow(Pi,2)/6+2*pow(Euler,2)) 
164                 + x*(-4*pow(Euler, 3)/3 -pow(Pi,2)*Euler/3 - 2*zeta(3)/3) + Order(pow(x, 2));
165         result += check_series(e, 0, d);
166         
167         return result;
168 }
169
170 // Order term handling
171 static unsigned exam_series5()
172 {
173         unsigned result = 0;
174         ex e, d;
175
176         e = 1 + x + pow(x, 2) + pow(x, 3);
177         d = Order(1);
178         result += check_series(e, 0, d, 0);
179         d = 1 + Order(x);
180         result += check_series(e, 0, d, 1);
181         d = 1 + x + Order(pow(x, 2));
182         result += check_series(e, 0, d, 2);
183         d = 1 + x + pow(x, 2) + Order(pow(x, 3));
184         result += check_series(e, 0, d, 3);
185         d = 1 + x + pow(x, 2) + pow(x, 3);
186         result += check_series(e, 0, d, 4);
187         return result;
188 }
189
190 // Series expansion of tgamma(-1)
191 static unsigned exam_series6()
192 {
193         ex e = tgamma(2*x);
194         ex d = pow(x+1,-1)*numeric(1,4) +
195                pow(x+1,0)*(numeric(3,4) -
196                            numeric(1,2)*Euler) +
197                pow(x+1,1)*(numeric(7,4) -
198                            numeric(3,2)*Euler +
199                            numeric(1,2)*pow(Euler,2) +
200                            numeric(1,12)*pow(Pi,2)) +
201                pow(x+1,2)*(numeric(15,4) -
202                            numeric(7,2)*Euler -
203                            numeric(1,3)*pow(Euler,3) +
204                            numeric(1,4)*pow(Pi,2) +
205                            numeric(3,2)*pow(Euler,2) -
206                            numeric(1,6)*pow(Pi,2)*Euler -
207                            numeric(2,3)*zeta(3)) +
208                pow(x+1,3)*(numeric(31,4) - pow(Euler,3) -
209                            numeric(15,2)*Euler +
210                            numeric(1,6)*pow(Euler,4) +
211                            numeric(7,2)*pow(Euler,2) +
212                            numeric(7,12)*pow(Pi,2) -
213                            numeric(1,2)*pow(Pi,2)*Euler -
214                            numeric(2)*zeta(3) +
215                            numeric(1,6)*pow(Euler,2)*pow(Pi,2) +
216                            numeric(1,40)*pow(Pi,4) +
217                            numeric(4,3)*zeta(3)*Euler) +
218                Order(pow(x+1,4));
219         return check_series(e, -1, d, 4);
220 }
221         
222 // Series expansion of tan(x==Pi/2)
223 static unsigned exam_series7()
224 {
225         ex e = tan(x*Pi/2);
226         ex d = pow(x-1,-1)/Pi*(-2) + pow(x-1,1)*Pi/6 + pow(x-1,3)*pow(Pi,3)/360
227               +pow(x-1,5)*pow(Pi,5)/15120 + pow(x-1,7)*pow(Pi,7)/604800
228               +Order(pow(x-1,9));
229         return check_series(e,1,d,9);
230 }
231
232 // Series expansion of log(sin(x==0))
233 static unsigned exam_series8()
234 {
235         ex e = log(sin(x));
236         ex d = log(x) - pow(x,2)/6 - pow(x,4)/180 - pow(x,6)/2835 - pow(x,8)/37800 + Order(pow(x,9));
237         return check_series(e,0,d,9);
238 }
239
240 // Series expansion of Li2(sin(x==0))
241 static unsigned exam_series9()
242 {
243         ex e = Li2(sin(x));
244         ex d = x + pow(x,2)/4 - pow(x,3)/18 - pow(x,4)/48
245                - 13*pow(x,5)/1800 - pow(x,6)/360 - 23*pow(x,7)/21168
246                + Order(pow(x,8));
247         return check_series(e,0,d,8);
248 }
249
250 // Series expansion of Li2((x==2)^2), caring about branch-cut
251 static unsigned exam_series10()
252 {
253         using GiNaC::log;
254
255         ex e = Li2(pow(x,2));
256         ex d = Li2(4) + (-log(3) + I*Pi*csgn(I-I*pow(x,2))) * (x-2)
257                + (numeric(-2,3) + log(3)/4 - I*Pi/4*csgn(I-I*pow(x,2))) * pow(x-2,2)
258                + (numeric(11,27) - log(3)/12 + I*Pi/12*csgn(I-I*pow(x,2))) * pow(x-2,3)
259                + (numeric(-155,648) + log(3)/32 - I*Pi/32*csgn(I-I*pow(x,2))) * pow(x-2,4)
260                + Order(pow(x-2,5));
261         return check_series(e,2,d,5);
262 }
263
264 // Series expansion of logarithms around branch points
265 static unsigned exam_series11()
266 {
267         using GiNaC::log;
268
269         unsigned result = 0;
270         ex e, d;
271         symbol a("a");
272         
273         e = log(x);
274         d = log(x);
275         result += check_series(e,0,d,5);
276         
277         e = log(3/x);
278         d = log(3)-log(x);
279         result += check_series(e,0,d,5);
280         
281         e = log(3*pow(x,2));
282         d = log(3)+2*log(x);
283         result += check_series(e,0,d,5);
284         
285         // These ones must not be expanded because it would result in a branch cut
286         // running in the wrong direction. (Other systems tend to get this wrong.)
287         e = log(-x);
288         d = e;
289         result += check_series(e,0,d,5);
290         
291         e = log(I*(x-123));
292         d = e;
293         result += check_series(e,123,d,5);
294         
295         e = log(a*x);
296         d = e;  // we don't know anything about a!
297         result += check_series(e,0,d,5);
298         
299         e = log((1-x)/x);
300         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));
301         result += check_series(e,1,d,5);
302         
303         return result;
304 }
305
306 // Series expansion of other functions around branch points
307 static unsigned exam_series12()
308 {
309         using GiNaC::log;
310
311         unsigned result = 0;
312         ex e, d;
313         
314         // NB: Mma and Maple give different results, but they agree if one
315         // takes into account that by assumption |x|<1.
316         e = atan(x);
317         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));
318         result += check_series(e,I,d,3);
319         
320         // NB: here, at -I, Mathematica disagrees, but it is wrong -- they
321         // pick up a complex phase by incorrectly expanding logarithms.
322         e = atan(x);
323         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));
324         result += check_series(e,-I,d,3);
325         
326         // This is basically the same as above, the branch point is at +/-1:
327         e = atanh(x);
328         d = (-log(2)/2+log(x+1)/2) + (x+1)/4 + pow(x+1,2)/16 + Order(pow(x+1,3));
329         result += check_series(e,-1,d,3);
330         
331         return result;
332 }
333
334
335 unsigned exam_pseries()
336 {
337         unsigned result = 0;
338         
339         cout << "examining series expansion" << flush;
340         clog << "----------series expansion:" << endl;
341         
342         result += exam_series1();  cout << '.' << flush;
343         result += exam_series2();  cout << '.' << flush;
344         result += exam_series3();  cout << '.' << flush;
345         result += exam_series4();  cout << '.' << flush;
346         result += exam_series5();  cout << '.' << flush;
347         result += exam_series6();  cout << '.' << flush;
348         result += exam_series7();  cout << '.' << flush;
349         result += exam_series8();  cout << '.' << flush;
350         result += exam_series9();  cout << '.' << flush;
351         result += exam_series10();  cout << '.' << flush;
352         result += exam_series11();  cout << '.' << flush;
353         result += exam_series12();  cout << '.' << flush;
354         
355         if (!result) {
356                 cout << " passed " << endl;
357                 clog << "(no output)" << endl;
358         } else {
359                 cout << " failed " << endl;
360         }
361         return result;
362 }