source: inundation/ga/storm_surge/pyvolution/test_least_squares.py @ 305

#!/usr/bin/env python

import unittest
from math import sqrt

from least_squares import *
from config import epsilon
from Numeric import allclose, array

def distance(x, y):
    return sqrt( sum( (array(x)-array(y))**2 ))         
        
class TestCase(unittest.TestCase):
    def setUp(self):
        pass
        
    def tearDown(self):
        pass


    def test_datapoint_at_centroid(self):
        a = [0.0, 0.0]
        b = [0.0, 2.0]
        c = [2.0,0.0]
        points = [a, b, c]
        triangles = [ [1,0,2] ]   #bac

        data = [ [2.0/3, 2.0/3] ] #Use centroid as one data point        
        
        mesh = Interpolation(points, triangles, data)
        assert allclose(mesh.matrix, [[1./3, 1./3, 1./3]]) 
        


    def test_datapoints_at_vertices(self):
        """Test that data points coinciding with vertices yield a diagonal matrix
        """
        
        a = [0.0, 0.0]
        b = [0.0, 2.0]
        c = [2.0,0.0]
        points = [a, b, c]
        triangles = [ [1,0,2] ]   #bac

        data = points #Use data at vertices        
        
        mesh = Interpolation(points, triangles, data)
        assert allclose(mesh.matrix, [[1., 0., 0.],
                                      [0., 1., 0.],
                                      [0., 0., 1.]]) 
        


    def test_datapoints_on_edge_midpoints(self):
        """Try datapoints midway on edges -
        each point should affect two matrix entries equally
        """
        
        a = [0.0, 0.0]
        b = [0.0, 2.0]
        c = [2.0,0.0]
        points = [a, b, c]
        triangles = [ [1,0,2] ]   #bac

        data = [ [0., 1.], [1., 0.], [1., 1.] ]
        
        mesh = Interpolation(points, triangles, data)
        
        assert allclose(mesh.matrix, [[0.5, 0.5, 0.0],  #Affects vertex 1 and 0       
                                      [0.5, 0.0, 0.5],  #Affects vertex 0 and 2
                                      [0.0, 0.5, 0.5]]) #Affects vertex 1 and 2


    def test_datapoints_on_edges(self):
        """Try datapoints on edges -
        each point should affect two matrix entries in proportion
        """
        
        a = [0.0, 0.0]
        b = [0.0, 2.0]
        c = [2.0,0.0]
        points = [a, b, c]
        triangles = [ [1,0,2] ]   #bac

        data = [ [0., 1.5], [1.5, 0.], [1.5, 0.5] ]
        
        mesh = Interpolation(points, triangles, data)
        
        assert allclose(mesh.matrix, [[0.25, 0.75, 0.0],  #Affects vertex 1 and 0      
                                      [0.25, 0.0, 0.75],  #Affects vertex 0 and 2
                                      [0.0, 0.25, 0.75]]) #Affects vertex 1 and 2

    def test_arbitrary_datapoints(self):
        """Try arbitrary datapoints
        """

        from Numeric import sum
        
        a = [0.0, 0.0]
        b = [0.0, 2.0]
        c = [2.0,0.0]
        points = [a, b, c]
        triangles = [ [1,0,2] ]   #bac

        data = [ [0.2, 1.5], [0.123, 1.768], [1.43, 0.44] ]
        
        mesh = Interpolation(points, triangles, data)

        assert allclose(sum(mesh.matrix, axis=1), 1.0)
        

           
    def test_more_triangles(self):


        #FIXME: Not finished yet
        a = [0.0, 0.0]
        b = [0.0, 2.0]
        c = [2.0,0.0]
        d = [0.0, 4.0]
        e = [2.0, 2.0]
        f = [4.0,0.0]

        points = [a, b, c, d, e, f]
        #bac, bce, ecf, dbe, daf, dae 
        triangles = [ [1,0,2], [1,2,4], [4,2,5], [3,1,4], [3,0,5], [3,0,4]]


        #Data points
        data = [ [0., 2.0], [1.15, 0.75], [0.8888, 0.21], [1.000001, 0.00001] ]

        mesh = Interpolation(points, triangles, data)        

        #print mesh.matrix
       

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