source: anuga_core/source/anuga/shallow_water/test_most2nc.py @ 5541

Last change on this file since 5541 was 3567, checked in by ole, 18 years ago

More heavy handed restructuring towards the agreed formats.

File size: 1.1 KB
Line 
1import unittest
2from Numeric import allclose, asarray
3import most2nc
4from Scientific.IO.NetCDF import NetCDFFile
5import os
6
7FN = 'small___.txt'
8
9class Test_most2nc(unittest.TestCase):
10    def setUp(self):
11        fid = open(FN, 'w')
12        fid.write("""4 4
13150.66667
14150.83334
15151.
16151.16667
17-34.
18-34.16667
19-34.33333
20-34.5
21-1. -2. -3. -4.
22-5. -6. -7. -8.
23-9. -10. -11. -12.
24-13. -14. -15. -16.
25""")
26        fid.close()
27                 
28    def tearDown(self):
29        os.remove(FN)
30
31    def test_small_nxn(self):
32        most2nc.most2nc(input_file=FN,output_file='test.nc'\
33                        ,inverted_bathymetry = False,verbose = False)
34
35        fid = NetCDFFile('test.nc')
36        elevation = fid.variables['ELEVATION'][:]
37        fid.close()
38
39        z=[[-13., -14., -15., -16.]\
40           ,[-9., -10., -11., -12.]\
41           ,[-5.,  -6.,  -7.,  -8.]\
42           ,[-1.,  -2.,  -3.,  -4.]]
43        z = asarray(z)
44
45        assert allclose(z,elevation)
46        import os
47        os.remove('test.nc')
48       
49if __name__ == "__main__":
50    suite = unittest.makeSuite(Test_most2nc,'test')
51    runner = unittest.TextTestRunner()
52    runner.run(suite)
Note: See TracBrowser for help on using the repository browser.