-
Notifications
You must be signed in to change notification settings - Fork 303
/
t003_test.py
118 lines (88 loc) · 3.08 KB
/
t003_test.py
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
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
import flopy
import os
from nose.tools import raises
def test_loadfreyberg():
cwd = os.getcwd()
pth = os.path.join('..', 'examples', 'data', 'freyberg')
print(pth)
assert (os.path.isdir(pth))
os.chdir(pth)
namefile = 'freyberg.nam'
ml = flopy.modflow.Modflow.load(namefile, verbose=True)
os.chdir(cwd)
assert isinstance(ml, flopy.modflow.Modflow)
assert ml.load_fail is False
return
def test_loadoahu():
cwd = os.getcwd()
pth = os.path.join('..', 'examples', 'data', 'parameters')
assert (os.path.isdir(pth))
os.chdir(pth)
namefile = 'Oahu_01.nam'
ml = flopy.modflow.Modflow.load(namefile, verbose=True)
os.chdir(cwd)
assert isinstance(ml, flopy.modflow.Modflow)
assert ml.load_fail is False
return
def test_loadtwrip():
cwd = os.getcwd()
pth = os.path.join('..', 'examples', 'data', 'parameters')
assert (os.path.isdir(pth))
os.chdir(pth)
namefile = 'twrip.nam'
ml = flopy.modflow.Modflow.load(namefile, verbose=True)
os.chdir(cwd)
assert isinstance(ml, flopy.modflow.Modflow)
assert ml.load_fail is False
return
def test_loadoc():
ws = os.path.join('temp', 't003')
ml = flopy.modflow.Modflow(model_ws=ws)
fpth = os.path.join('..', 'examples', 'data', 'mf2005_test', 'fhb.dis')
dis = flopy.modflow.ModflowDis.load(fpth, ml, check=False)
fpth = os.path.join('..', 'examples', 'data', 'mf2005_test', 'fhb.oc')
oc = flopy.modflow.ModflowOc.load(fpth, ml, ext_unit_dict=None)
return
@raises(IOError)
def test_loadoc_lenfail():
ws = os.path.join('temp', 't003')
ml = flopy.modflow.Modflow(model_ws=ws)
fpth = os.path.join('..', 'examples', 'data', 'mf2005_test', 'fhb.oc')
oc = flopy.modflow.ModflowOc.load(fpth, ml, nper=3, nstp=1, nlay=1)
return
@raises(ValueError)
def test_loadoc_nperfail():
ws = os.path.join('temp', 't003')
ml = flopy.modflow.Modflow(model_ws=ws)
fpth = os.path.join('..', 'examples', 'data', 'mf2005_test', 'fhb.oc')
oc = flopy.modflow.ModflowOc.load(fpth, ml, nper=0, nlay=1)
return
@raises(ValueError)
def test_loadoc_nlayfail():
ws = os.path.join('temp', 't003')
ml = flopy.modflow.Modflow(model_ws=ws)
fpth = os.path.join('..', 'examples', 'data', 'mf2005_test', 'fhb.oc')
oc = flopy.modflow.ModflowOc.load(fpth, ml, nper=3, nlay=0)
return
@raises(ValueError)
def test_loadoc_nstpfail():
ws = os.path.join('temp', 't003')
ml = flopy.modflow.Modflow(model_ws=ws)
fpth = os.path.join('..', 'examples', 'data', 'mf2005_test', 'fhb.oc')
oc = flopy.modflow.ModflowOc.load(fpth, ml, nper=3, nlay=1)
return
@raises(IOError)
def test_load_nam_mf_nonexistant_file():
ml = flopy.modflow.Modflow.load('nonexistant.nam')
@raises(IOError)
def test_load_nam_mt_nonexistant_file():
ml = flopy.mt3d.Mt3dms.load('nonexistant.nam')
if __name__ == '__main__':
test_loadoc()
test_loadoc_nstpfail()
test_load_nam_mf_nonexistant_file()
test_load_nam_mt_nonexistant_file()
test_loadoc_lenfail()
test_loadfreyberg()
test_loadoahu()
test_loadtwrip()