def test_getitem(self):
m = GMS(testmodel, verbosity=-1)
m.layer_add_var('T')
w = m.get_well(0, 0, 'T')
self.assertListEqual(w['T'].tolist(), [0., 200., 800., 1000.])
def test_plot_grad(self):
m = GMS(testmodel, verbosity=-1)
m.layer_add_var('T')
w = m.get_well(0, 0, 'T')
for grad in w.plot_grad('T', return_array=True, absolute=True):
self.assertAlmostEqual(grad, 0.02)
def test_call(self):
m = GMS(testmodel, verbosity=-1)
w = m.get_well(0, 0 )
self.assertListEqual(w().tolist(), [0., -10e3, -40e3, -50e3])