def test_lgrutil():
nlayp = 5
nrowp = 5
ncolp = 5
delrp = 100.
delcp = 100.
topp = 100.
botmp = [-100, -200, -300, -400, -500]
idomainp = np.ones((nlayp, nrowp, ncolp), dtype=np.int)
idomainp[0:2, 1:4, 1:4] = 0
ncpp = 3
ncppl = [1, 1, 0, 0, 0]
lgr = Lgr(nlayp,
nrowp,
ncolp,
delrp,
delcp,
topp,
botmp,
idomainp,
ncpp=ncpp,
ncppl=ncppl,
xllp=100.,
yllp=100.)
# child shape
assert lgr.get_shape() == (2, 9, 9), 'child shape is not (2, 9, 9)'
# child delr/delc
delr, delc = lgr.get_delr_delc()
assert np.allclose(delr, delrp / ncpp), 'child delr not correct'
assert np.allclose(delc, delcp / ncpp), 'child delc not correct'
# child idomain
idomain = lgr.get_idomain()
assert idomain.min() == idomain.max() == 1
assert idomain.shape == (2, 9, 9)
# replicated parent array
ap = np.arange(nrowp * ncolp).reshape((nrowp, ncolp))
ac = lgr.get_replicated_parent_array(ap)
assert ac[0, 0] == 6
assert ac[-1, -1] == 18
# child top/bottom
topc, botmc = lgr.get_top_botm()
assert topc.shape == (9, 9)
assert botmc.shape == (2, 9, 9)
assert topc.min() == topc.max() == 100.
errmsg = '{} /= {}'.format(botmc[:, 0, 0], np.array(botmp[:2]))
assert np.allclose(botmc[:, 0, 0], np.array(botmp[:2])), errmsg
# exchange data
exchange_data = lgr.get_exchange_data(angldegx=True, cdist=True)
ans1 = [(0, 1, 0), (0, 0, 0), 1, 50.0, 16.666666666666668,
33.333333333333336, 0.0, 354.33819375782156]
errmsg = '{} /= {}'.format(ans1, exchange_data[0])
assert exchange_data[0] == ans1, errmsg
ans2 = [(2, 3, 3), (1, 8, 8), 0, 50.0, 50, 1111.1111111111113, 180., 100.]
errmsg = '{} /= {}'.format(ans2, exchange_data[-1])
assert exchange_data[-1] == ans2, errmsg
errmsg = 'exchanges should be 71 horizontal plus 81 vertical'
assert len(exchange_data) == 72 + 81, errmsg
# list of parent cells connected to a child cell
assert lgr.get_parent_connections(0, 0, 0) == [((0, 1, 0), -1),
((0, 0, 1), 2)]
assert lgr.get_parent_connections(1, 8, 8) == [((1, 3, 4), 1),
((1, 4, 3), -2),
((2, 3, 3), -3)]
return
def test_lgrutil():
nlayp = 5
nrowp = 5
ncolp = 5
delrp = 100.
delcp = 100.
topp = 100.
botmp = [-100, -200, -300, -400, -500]
idomainp = np.ones((nlayp, nrowp, ncolp), dtype=np.int)
idomainp[0:2, 1:4, 1:4] = 0
ncpp = 3
ncppl = [1, 1, 0, 0, 0]
lgr = Lgr(nlayp,
nrowp,
ncolp,
delrp,
delcp,
topp,
botmp,
idomainp,
ncpp=ncpp,
ncppl=ncppl,
xllp=100.,
yllp=100.)
# child shape
assert lgr.get_shape() == (2, 9, 9), 'child shape is not (2, 9, 9)'
# child delr/delc
delr, delc = lgr.get_delr_delc()
assert np.allclose(delr, delrp / ncpp), 'child delr not correct'
assert np.allclose(delc, delcp / ncpp), 'child delc not correct'
# child idomain
idomain = lgr.get_idomain()
assert idomain.min() == idomain.max() == 1
assert idomain.shape == (2, 9, 9)
# replicated parent array
ap = np.arange(nrowp * ncolp).reshape((nrowp, ncolp))
ac = lgr.get_replicated_parent_array(ap)
assert ac[0, 0] == 6
assert ac[-1, -1] == 18
# top/bottom
top, botm = lgr.get_top_botm()
assert top.shape == (9, 9)
assert botm.shape == (2, 9, 9)
assert top.min() == top.max() == 100.
assert botm.min() == botm.max() == 0.
# exchange data
exchange_data = lgr.get_exchange_data(angldegx=True, cdist=True)
ans1 = [(0, 1, 0), (0, 0, 0), 1, 50.0, 16.666666666666668,
33.333333333333336, 0.0, 354.33819375782156]
ans2 = [(1, 4, 3), (1, 8, 8), 1, 50.0, 16.666666666666668,
33.333333333333336, 90.0, 354.3381937578216]
assert exchange_data[0] == ans1
assert exchange_data[-1] == ans2
assert len(exchange_data) == 72
# list of parent cells connected to a child cell
assert lgr.get_parent_connections(0, 0, 0) == [((0, 1, 0), -1),
((0, 0, 1), 2)]
assert lgr.get_parent_connections(1, 8, 8) == [((1, 3, 4), 1),
((1, 4, 3), -2)]
return