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])