Changeset 5494 for anuga_work/development/Hinwood_2008/test_slope.py

Timestamp:

Jul 11, 2008, 4:15:59 PM (15 years ago)

Author:

duncan

Message:

Current Hinwood scenario - added plotting of froude number

File:

• anuga_work/development/Hinwood_2008/test_slope.py (modified) (3 diffs)

anuga_work/development/Hinwood_2008/test_slope.py

@@ -10 +10 @@
 import tempfile
 
-from scipy import array, allclose
+from Numeric import array, allclose
+from anuga.utilities.numerical_tools import ensure_numeric
+
 
 from slope import *
@@ -33 +35 @@
         handle.write(sample)
         handle.close()
-        dtimes, depths, sensors = load_depths(file_name)
+        dtimes, depths, sensors = load_sensors(file_name)
         actual = array([0,0.01,0.02,0.03])
         self.assert_ (allclose(dtimes, actual, 0.001))
@@ -73 +75 @@
         handle.write(sample)
         handle.close()
-        graph_slopes(file_name)
 
+        # since it currently graphs 
+        #graph_slopes(file_name)
         
         os.remove(file_name)
               
+    def test_get_min_in_band(self):
+        time = array([0,1,2,3,4,5])
+        slopes = array([[10,12],
+                        [9,10],
+                        [8,9],
+                        [9,7],
+                        [10,6],
+                        [7,2]])
+
+        max_q, max_q_times = get_min_in_band(0,5,time,slopes)
+        actual = array([8,6])
+        self.assert_ (allclose(max_q, actual, 0.001))
+        actual = array([2,4])
+        self.assert_ (allclose(max_q_times, actual, 0.001))
+
+        
+        max_q, max_q_times = get_min_in_band(1,4,time,slopes)
+        actual = array([8,7])
+        self.assert_ (allclose(max_q, actual, 0.001))
+        actual = array([2,3])
+        self.assert_ (allclose(max_q_times, actual, 0.001))
+        
+    def test_break_times2bands(self):
+        break_times = [1,3,4,8]
+        band_offset = 0.0
+        bands  = break_times2bands(break_times, band_offset)
+        #print "bands", bands
+        actual = array([0,2,3.5,6,10 ])
+        self.assert_ (allclose(ensure_numeric(bands), actual, 0.001))
+        
+    def test_get_time_band(self):
+        time = array([0,1,2,3,4,5])
+        slopes = array([[10,12],
+                        [9,10],
+                        [8,9],
+                        [9,7],
+                        [10,6],
+                        [7,2]])
+
+        band_time, band_quantity = get_time_band(0,5,time,slopes)
+        actual = array([1,2,3,4])
+        self.assert_ (allclose(band_time, actual, 0.001))
+        actual = array([[9,10],
+                        [8,9],
+                        [9,7],
+                        [10,6]])
+        self.assert_ (allclose(band_quantity, actual, 0.001))
+
+    
+    def test_get_band(self):
+        time = array([0,1,2,3,4,5])
+        slopes = array([[10,12],
+                        [9,10],
+                        [8,9],
+                        [9,7],
+                        [10,6],
+                        [7,2]])
+
+        band_time, band_quantity = get_band(0,5,time,slopes,0)
+        actual = array([1,2,3,4])
+        self.assert_ (allclose(band_time, actual, 0.001))
+        actual = array([[9,10],
+                        [8,9],
+                        [9,7],
+                        [10,6]])
+        self.assert_ (allclose(band_quantity, actual, 0.001))
+        
+        location = array([0,1,2,3,4,5])
+        slopes = array([[0,1,2,3,4,5],
+                        [0,10,20,30,40,50]])
+
+        band_time, band_quantity = get_band(0,5,location,slopes,-1)
+        actual = array([1,2,3,4])
+        self.assert_ (allclose(band_time, actual, 0.001))
+        actual = array([[1,2,3,4],
+                        [10,20,30,40]])
+        self.assert_ (allclose(band_quantity, actual, 0.001))      
+        
 if __name__ == "__main__":
 
     suite = unittest.makeSuite(slopeTestCase,'test')
+    suite = unittest.makeSuite(slopeTestCase,'test_get_band')
+
     runner = unittest.TextTestRunner()  #verbosity=2)
     runner.run(suite)