source: anuga_work/development/Hinwood_2008/test_slope.py @ 5447

#!/usr/bin/env python
#
"""
I removed lone test vert's, since I'm working on removing lone verts at a lower
level of the code, using the -j flag in triangle.
"""
import sys
import os
import unittest
import tempfile

from scipy import array, allclose

from slope import *

class slopeTestCase(unittest.TestCase):
    def setUp(self):
        pass

    def tearDown(self):
        pass

    def test_slope_from_file(self):
        handle, file_name = tempfile.mkstemp('.csv', __name__+'_')
        os.close(handle)
        handle = open(file_name,'w')
        
        sample = """time,0.5:0.5,0.501:0.5,0.502:0.5
0,0.1,0.2,0.3
0.01,0.1,0.2,1.2
0.02,0.1,0.1,0.0
0.03,0.5,0.501,0.502"""
        handle.write(sample)
        handle.close()
        dtimes, depths, sensors = load_depths(file_name)
        actual = array([0,0.01,0.02,0.03])
        self.assert_ (allclose(dtimes, actual, 0.001))
        actual = array([0.5, 0.501, 0.502])
        self.assert_ (allclose(depths, actual, 0.001))

        actual = array([[ 0.1,    0.2,    0.3  ],
                        [ 0.1 ,   0.2  ,  1.2  ],
                        [ 0.1 ,   0.1,    0.   ],
                        [ 0.5 ,   0.501 , 0.502]])
        self.assert_ (allclose(sensors, actual, 0.001))

        times, slope_locations, slopes  = load_slopes(file_name)
        actual = array([0.5005,  0.5015])
        self.assert_ (allclose(slope_locations, actual, 0.001))
        actual = array([0,0.01,0.02,0.03])
        self.assert_ (allclose(times, actual, 0.001))

        actual = array([[ 100.,  100.],
                        [ 100., 1000.],
                        [   0., -100.],
                        [   1.,    1.]])
        self.assert_ (allclose(slopes, actual, 0.001))

        
        os.remove(file_name)
  
    def test_plot_from_file(self):
        handle, file_name = tempfile.mkstemp('.csv', __name__+'_')
        os.close(handle)
        handle = open(file_name,'w')
        
        sample = """time,0.0:1,1:0.5,2:0.5,3:0.5,4:0.5,5:0.5,6:0.5,7:0.5
0, 0, 1, 2, 1, 0, 0, 0, 0
1, 0, 0, 1, 2, 1, 0, 0, 0
2, 0, 0, 0, 1.5, 3, 1.5, 0, 0
3, 0, 0, 0, 0, 1.5, 3, 0, 0
4, 0, 0, 0, 0, 0, 1.5, 3, 0"""
        handle.write(sample)
        handle.close()
        graph_slopes(file_name)

        
        os.remove(file_name)
              
if __name__ == "__main__":

    suite = unittest.makeSuite(slopeTestCase,'test')
    runner = unittest.TextTestRunner()  #verbosity=2)
    runner.run(suite)