Newer
Older
'''
Created on 05/11/2015
@author: MMPE
'''
import unittest
testfilepath = 'test_files/'
class Test(unittest.TestCase):
def test_shearfile(self):
f = testfilepath + "tmp_shearfile.dat"
shear_file.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)
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
def test_shearfile2(self):
f = testfilepath + "tmp_shearfile.dat"
shear_file.save(f, [-55, 55], [30, 100, 160] , u=np.array([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)
def test_shear_makedirs(self):
f = testfilepath + "shear/tmp_shearfile2.dat"
shear_file.save(f, [-55, 55], [30, 100, 160] , u=np.array([0.7, 1, 1.3]).T)
shutil.rmtree(testfilepath + "shear")
if __name__ == "__main__":
#import sys;sys.argv = ['', 'Test.test_shearfile']
unittest.main()