Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
'''
Created on 05/11/2015
@author: MMPE
'''
import unittest
from wetb.hawc2 import shearfile
import numpy as np
import os
testfilepath = 'test_files/'
class Test(unittest.TestCase):
def test_shearfile(self):
f = testfilepath + "tmp_shearfile.dat"
shearfile.save(f, [-55, 55], [30, 100, 160] , u=np.array([[0.7, 1, 1.3], [0.7, 1, 1.3]]).T)
with open(f) as fid:
self.assertEqual(fid.read(),
""" # autogenerated shear file
2 3
# shear v component
0.0000000000 0.0000000000
0.0000000000 0.0000000000
0.0000000000 0.0000000000
# shear u component
0.7000000000 0.7000000000
1.0000000000 1.0000000000
1.3000000000 1.3000000000
# shear w component
0.0000000000 0.0000000000
0.0000000000 0.0000000000
0.0000000000 0.0000000000
# y coordinates
-55.0000000000
55.0000000000
# z coordinates
30.0000000000
100.0000000000
160.0000000000
""")
os.remove(f)
if __name__ == "__main__":
#import sys;sys.argv = ['', 'Test.test_shearfile']
unittest.main()