source: inundation/ga/storm_surge/pyvolution/test_cg_solve.py @ 433

#!/usr/bin/env python

import unittest
#from math import sqrt

from Numeric import allclose, array, transpose

from cg_solve import *

from scipy import sparse
        
class TestCase(unittest.TestCase):

    def setUp(self):
        pass
        
    def tearDown(self):
        pass


    def test_solve(self):
        A = [[2.0, -1.0, 0.0, 0.0 ],
             [-1.0, 2.0, -1.0, 0.0],
             [0.0, -1.0, 2.0, -1.0],
             [0.0,0.0, -1.0, 2.0]]
        
        A = sparse.dok_matrix(A)
        
        xe = [0.0, 1.0, 2.0, 3.0]
        b  = A*xe
        x =  [0.0, 0.0, 0.0, 0.0]

        x = conjugate_gradient(A,b,x)

        print "x",x

        assert allclose(x,xe)


    def test_solve_more(self):
        A = [[2.0, -1.0, 0.0, 0.0 ],
             [-1.0, 2.0, -1.0, 0.0],
             [0.0, -1.0, 2.0, -1.0],
             [0.0,0.0, -1.0, 2.0]]
        
        A = sparse.dok_matrix(A)
        
        xe = [[0.0, 1.0], [1.0, 3.0], [2.0, 5.0], [3.0, 9.0]]
        b  = A*xe
        x =  [[0.0, 0.0], [0.0, 0.0],[0.0, 0.0],[0.0, 0.0]]

        x = conjugate_gradient(A,b,x)

        print "x",x

        assert allclose(x,xe)

        
#-------------------------------------------------------------
if __name__ == "__main__":
    suite = unittest.makeSuite(TestCase,'test')
    runner = unittest.TextTestRunner()
    runner.run(suite)