Changeset 623 for inundation/ga/storm_surge/pyvolution/test_util.py

Timestamp:

Nov 24, 2004, 4:43:40 PM (20 years ago)

Author:

ole

Message:

Created and tested general File_function (reading and interpolating time series from file) and used it to refactor and simplify File_boundary

File:

• inundation/ga/storm_surge/pyvolution/test_util.py (modified) (1 diff)

inundation/ga/storm_surge/pyvolution/test_util.py

@@ -138 +138 @@
 
 
+
+    def test_file_function_time(self):
+        """Test that File function interpolates correctly
+        between given times. No x,y dependency here.
+        """
+
+        #Write file
+        import os, time
+        from config import time_format
+        from math import sin, pi
+
+        finaltime = 1200
+        filename = 'test_file_function.txt'
+        fid = open(filename, 'w')
+        start = time.mktime(time.strptime('2000', '%Y'))
+        dt = 60  #One minute intervals
+        t = 0.0
+        while t <= finaltime:
+            t_string = time.strftime(time_format, time.gmtime(t+start))
+            fid.write('%s, %f %f %f\n' %(t_string, 2*t, t**2, sin(t*pi/600)))
+            t += dt
+    
+        fid.close()
+
+        F = File_function(filename)
+
+        #Now try interpolation
+        for i in range(20):
+            t = i*10
+            q = F(t)
+
+            #Exact linear intpolation
+            assert allclose(q[0], 2*t) 
+            if i%6 == 0:
+                assert allclose(q[1], t**2)
+                assert allclose(q[2], sin(t*pi/600))                
+
+        #Check non-exact
+
+        t = 90 #Halfway between 60 and 120
+        q = F(t) 
+        assert allclose( (120**2 + 60**2)/2, q[1] )
+        assert allclose( (sin(120*pi/600) + sin(60*pi/600))/2, q[2] )
+
+
+        t = 100 #Two thirds of the way between between 60 and 120
+        q = F(t) 
+        assert allclose( 2*120**2/3 + 60**2/3, q[1] )
+        assert allclose( 2*sin(120*pi/600)/3 + sin(60*pi/600)/3, q[2] )
+        
+
+        
+
+    def test_file_function_time_with_domain(self):
+        """Test that File function interpolates correctly
+        between given times. No x,y dependency here.
+        Use domain with starttime
+        """
+
+        #Write file
+        import os, time, calendar
+        from config import time_format
+        from math import sin, pi
+        from domain import Domain
+
+        finaltime = 1200
+        filename = 'test_file_function.txt'
+        fid = open(filename, 'w')
+        start = time.mktime(time.strptime('2000', '%Y'))
+        dt = 60  #One minute intervals
+        t = 0.0
+        while t <= finaltime:
+            t_string = time.strftime(time_format, time.gmtime(t+start))
+            fid.write('%s, %f %f %f\n' %(t_string, 2*t, t**2, sin(t*pi/600)))
+            t += dt
+   
+        fid.close()
+
+        a = [0.0, 0.0]
+        b = [4.0, 0.0]
+        c = [0.0, 3.0]
+       
+        points = [a, b, c]
+        vertices = [[0,1,2]]
+        domain = Domain(points, vertices) 
+
+        #Check that domain.starttime is updated if non-existing
+        F = File_function(filename, domain)
+        assert allclose(domain.starttime, start)        
+
+        #Check that domain.starttime is updated if too early
+        domain.starttime = start - 1                
+        F = File_function(filename, domain)
+        assert allclose(domain.starttime, start)
+
+        #Check that domain.starttime isn't updated if later
+        domain.starttime = start + 1                
+        F = File_function(filename, domain)
+        assert allclose(domain.starttime, start+1)                
+
+        
+       
+        #Now try interpolation
+        for i in range(20):
+            t = i*10
+            q = F(t)
+
+            #Exact linear intpolation
+            assert allclose(q[0], 2*t) 
+            if i%6 == 0:
+                assert allclose(q[1], t**2)
+                assert allclose(q[2], sin(t*pi/600))                
+
+        #Check non-exact
+
+        t = 90 #Halfway between 60 and 120
+        q = F(t) 
+        assert allclose( (120**2 + 60**2)/2, q[1] )
+        assert allclose( (sin(120*pi/600) + sin(60*pi/600))/2, q[2] )
+
+
+        t = 100 #Two thirds of the way between between 60 and 120
+        q = F(t) 
+        assert allclose( 2*120**2/3 + 60**2/3, q[1] )
+        assert allclose( 2*sin(120*pi/600)/3 + sin(60*pi/600)/3, q[2] )
+        
       
 #-------------------------------------------------------------