%              test script for SOR
% ------------------------------------------------------

fprintf('\n  ====== CAAM 454 SOR Testing ======\n')
format = ' Iter = %4i: rel_err = %8.4e,'; 
format = [format ' Cputime = %g\n\n']; 

N = input(' N = ');
omega = input(' omega = ');
A = gallery('poisson',N);
n = size(A,1);
b = A*ones(n,1);
fprintf('\n ----- Poisson Matrix, n = %g -----\n',n)  

% ------ run instructor's code ------
t0 = cputime;
[x, iter, info] = zSOR(A,b,omega,n,1.e-4);
rerr = norm(x-1)/sqrt(n);
fprintf(format,iter,rerr,cputime-t0);

% ------ run your code ------
if exist('mySOR','file');
    t0 = cputime;
    [x, iter, info] = mySOR(A,b,omega,n,1.e-4);
    rerr = norm(x-1)/sqrt(n);
    fprintf(format,iter,rerr,cputime-t0);
end