def setUp(self): M = Mesh.TensorMesh([np.ones(20)]) M.setCellGradBC('dirichlet') params = Richards.Empirical.HaverkampParams().celia1990 params['Ks'] = np.log(params['Ks']) E = Richards.Empirical.Haverkamp(M, **params) bc = np.array([-61.5, -20.7]) h = np.zeros(M.nC) + bc[0] prob = Richards.RichardsProblem(M, mapping=E, timeSteps=[(40, 3), (60, 3)], tolRootFinder=1e-6, debug=False, boundaryConditions=bc, initialConditions=h, doNewton=False, method='mixed') prob.Solver = Solver locs = np.r_[5., 10, 15] times = prob.times[3:5] rxSat = Richards.RichardsRx(locs, times, 'saturation') rxPre = Richards.RichardsRx(locs, times, 'pressureHead') survey = Richards.RichardsSurvey([rxSat, rxPre]) prob.pair(survey) self.h0 = h self.M = M self.Ks = params['Ks'] self.prob = prob self.survey = survey
def setUp(self): M = Mesh.TensorMesh([np.ones(8),np.ones(20),np.ones(10)]) M.setCellGradBC(['neumann','neumann','dirichlet']) params = Richards.Empirical.HaverkampParams().celia1990 params['Ks'] = np.log(params['Ks']) E = Richards.Empirical.Haverkamp(M, **params) bc = np.array([-61.5,-20.7]) bc = np.r_[np.zeros(M.nCy*M.nCz*2),np.zeros(M.nCx*M.nCz*2),np.ones(M.nCx*M.nCy)*bc[0],np.ones(M.nCx*M.nCy)*bc[1]] h = np.zeros(M.nC) + bc[0] prob = Richards.RichardsProblem(M,E, timeSteps=[(40,3),(60,3)], boundaryConditions=bc, initialConditions=h, doNewton=False, method='mixed', tolRootFinder=1e-6, debug=False) prob.Solver = Solver locs = Utils.ndgrid(np.r_[5,7.],np.r_[5,15.],np.r_[6,8.]) times = prob.times[3:5] rxSat = Richards.RichardsRx(locs, times, 'saturation') rxPre = Richards.RichardsRx(locs, times, 'pressureHead') survey = Richards.RichardsSurvey([rxSat, rxPre]) prob.pair(survey) self.h0 = h self.M = M self.Ks = params['Ks'] self.prob = prob self.survey = survey