source: inundation/ga/storm_surge/alpha_shape/test_alpha_shape.py @ 643

#!/usr/bin/env python

#TEST

import unittest
from Numeric import allclose

from alpha_shape import *

class TestCase(unittest.TestCase):

    def setUp(self):
        pass

         
    def tearDown(self):
        pass


    def test_alpha(self):
        a = [0.0, 0.0]
        b = [1.0, 0.0]
        c = [2.0, 0.0]
        d = [2.0, 2.0]
        e = [1.0, 2.0]
        f = [0.0, 2.0]

        alpha = Alpha_Shape([a,b,c,d,e,f])
        result = alpha.get_boundary()
        answer = [[0, 1], [1, 2], [2, 3], [3, 4], [4, 5], [5, 0]]
        #print "result",result
        assert allclose(answer, result) 

        
    def test_not_enough_points(self):
        a = [0.0, 0.0]
        b = [1.0, 0.0]

        try:
            alpha = Alpha_Shape([a,b])
        except PointError:
            pass
        else:
            self.failUnless(0==1,
                        'point list with 2 points did not raise an error!')


    def test_alpha_stand_alone(self):
        import os
        import tempfile
       
        fileName = tempfile.mktemp(".xya")
        file = open(fileName,"w")
        file.write("title row \n\
0.0, 0.0\n\
1.0, 0.0\n\
2.0, 0.0\n\
2.0, 2.0\n\
1.0, 2.0\n\
0.0, 2.0\n")
        file.close()

        output_file_name = tempfile.mktemp(".bnd")
        command = 'python .\\alpha_shape.py ' + fileName + ' ' + output_file_name
        #print "command", command
        os.system(command)
        os.remove(fileName)
        
        file = open(output_file_name,"r")
        lFile = file.read().split('\n')
        file.close()
        os.remove(output_file_name)
        self.failUnless(lFile[1] == "0,1" and
                        lFile[2] == "1,2" and 
                        lFile[3] == "2,3" and 
                        lFile[4] == "3,4" and 
                        lFile[5] == "4,5" and 
                        lFile[6] == "5,0" and 
                        lFile[7] == ""
                        ,
                        'boundary file is wrong')


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