Add unitary tests for arnoldi
[scilab.git] / scilab / modules / arnoldi / tests / unit_tests / znaupd.dia.ref
1 // =============================================================================
2 // Scilab ( http://www.scilab.org/ ) - This file is part of Scilab
3 // Copyright (C) 2012 - SE - Sylvestre Ledru
4 //
5 //  This file is distributed under the same license as the Scilab package.
6 // =============================================================================
7 nx    = 10;
8 nev   = 3;
9 ncv   = 6;
10 bmat  = 'I';
11 which = 'LM';
12 // Local Arrays
13 iparam  = zeros(11, 1);
14 ipntr   = zeros(14, 1);
15 _select = zeros(ncv, 1);
16 d       = zeros(nev + 1, 1) + 0 * %i;
17 z       = zeros(nx, nev) + 0* %i;
18 resid   = zeros(nx, 1) + 0 * %i; 
19 v       = zeros(nx, ncv) + 0 * %i;
20 workd   = zeros(3 * nx, 1) + 0 * %i; 
21 workev  = zeros(2 * ncv, 1) + 0 * %i;
22 rwork   = zeros(ncv, 1);
23 workl   = zeros(3 * ncv * ncv + 5 *ncv, 1) + 0 * %i;
24 // Build the complex test matrix
25 A            = diag(10 * ones(nx,1) + %i * ones(nx,1));
26 A(1:$-1,2:$) = A(1:$-1,2:$) + diag(6 * ones(nx - 1,1));
27 A(2:$,1:$-1) = A(2:$,1:$-1) + diag(-6 * ones(nx - 1,1));
28 tol    = 0;
29 ido    = 0;
30 ishfts = 1;
31 maxitr = 300;
32 mode1  = 1;
33 iparam(1) = ishfts;
34 iparam(3) = maxitr;
35 iparam(7) = mode1;
36 sigma = complex(0); 
37 info_znaupd = 0;
38 // M A I N   L O O P (Reverse communication)
39 while(ido <> 99)
40   // Repeatedly call the routine ZNAUPD and take actions indicated by parameter IDO until
41   // either convergence is indicated or maxitr has been exceeded.
42   [ido, resid, v, iparam, ipntr, workd, workl, info_znaupd] = znaupd(ido, bmat, nx, which, nev, tol, resid, ncv, v, iparam, ipntr, workd, workl, rwork, info_znaupd);
43   if(info_znaupd < 0)
44     printf('\nError with znaupd, info = %d\n', info_znaupd);
45     printf('Check the documentation of znaupd\n\n');
46   end
47   if(ido == -1 | ido == 1)
48     // Perform matrix vector multiplication 
49     workd(ipntr(2):ipntr(2) + nx - 1) = A * workd(ipntr(1):ipntr(1) + nx - 1);
50   end
51 end
52 // Post-Process using ZNEUPD.
53 rvec    = 1;
54 howmany = 'A';
55 info_zneupd = 0;
56 [d, z, resid, iparam, ipntr, workd, workl, rwork, info_zneupd] = zneupd(rvec, howmany, _select, d, z, sigma, workev, bmat, nx, which, nev, tol, resid, ncv, v, ...
57                                                                     iparam, ipntr, workd, workl, rwork, info_zneupd);
58 assert_checkalmostequal(A * z, z * diag(d(1:3)), sqrt(%eps), 1.e-10);