/[escript]/trunk/escript/py_src/test_linearPDEs.py
ViewVC logotype

Diff of /trunk/escript/py_src/test_linearPDEs.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

trunk/esys2/escript/py_src/test_linearPDEs.py revision 153 by jgs, Tue Oct 25 01:51:20 2005 UTC trunk/escript/py_src/test_linearPDEs.py revision 304 by gross, Fri Dec 2 06:04:06 2005 UTC
# Line 23  __date__="$Date$" Line 23  __date__="$Date$"
23    
24    
25    
26  from esys.escript.util import Lsup,kronecker,interpolate  from esys.escript.util import Lsup,kronecker,interpolate,whereZero
27  from esys.escript import Function,FunctionOnBoundary,FunctionOnContactZero,Solution,ReducedSolution,Vector,ContinuousFunction,Scalar  from esys.escript import Function,FunctionOnBoundary,FunctionOnContactZero,Solution,ReducedSolution,Vector,ContinuousFunction,Scalar
28  from esys.escript.linearPDEs import LinearPDE,IllegalCoefficientValue,Poisson  from esys.escript.linearPDEs import LinearPDE,IllegalCoefficientValue,Poisson
29  import numarray  import numarray
# Line 49  class Test_Poisson(Test_linearPDEs): Line 49  class Test_Poisson(Test_linearPDEs):
49      def test_setCoefficient_q(self):      def test_setCoefficient_q(self):
50          mypde=Poisson(self.domain,debug=self.DEBUG)          mypde=Poisson(self.domain,debug=self.DEBUG)
51          x=self.domain.getX()          x=self.domain.getX()
52          q_ref=interpolate(x[0].whereZero(),Solution(self.domain))          q_ref=interpolate(whereZero(x[0]),Solution(self.domain))
53          A_ref=kronecker(self.domain)          A_ref=kronecker(self.domain)
54          mypde.setValue(q=x[0].whereZero())          mypde.setValue(q=whereZero(x[0]))
55          self.failUnless(self.check(mypde.getCoefficientOfGeneralPDE("A"),A_ref),"A is not kronecker")          self.failUnless(self.check(mypde.getCoefficientOfGeneralPDE("A"),A_ref),"A is not kronecker")
56          self.failUnless(mypde.getCoefficientOfGeneralPDE("B").isEmpty(),"B is not empty")          self.failUnless(mypde.getCoefficientOfGeneralPDE("B").isEmpty(),"B is not empty")
57          self.failUnless(mypde.getCoefficientOfGeneralPDE("C").isEmpty(),"C is not empty")          self.failUnless(mypde.getCoefficientOfGeneralPDE("C").isEmpty(),"C is not empty")
# Line 93  class Test_Poisson(Test_linearPDEs): Line 93  class Test_Poisson(Test_linearPDEs):
93         #construct mask:         #construct mask:
94         msk=Scalar(0.,cf)         msk=Scalar(0.,cf)
95         for i in range(d):         for i in range(d):
96           msk+=x[i].whereZero()           msk+=whereZero(x[i])
97         #construct right hand side         #construct right hand side
98         f=Scalar(0,cf)         f=Scalar(0,cf)
99         for i in range(d):         for i in range(d):
# Line 333  class Test_LinearPDE(Test_linearPDEs): Line 333  class Test_LinearPDE(Test_linearPDEs):
333      def test_resetCoefficient_HomogeneousConstraint(self):      def test_resetCoefficient_HomogeneousConstraint(self):
334          mypde=LinearPDE(self.domain,debug=self.DEBUG)          mypde=LinearPDE(self.domain,debug=self.DEBUG)
335          x=self.domain.getX()          x=self.domain.getX()
336          mypde.setValue(A=kronecker(self.domain),Y=1.,q=x[0].whereZero())          mypde.setValue(A=kronecker(self.domain),Y=1.,q=whereZero(x[0]))
337          u1=mypde.getSolution()          u1=mypde.getSolution()
338          mypde.setValue(Y=2.)          mypde.setValue(Y=2.)
339          u2=mypde.getSolution()          u2=mypde.getSolution()
# Line 341  class Test_LinearPDE(Test_linearPDEs): Line 341  class Test_LinearPDE(Test_linearPDEs):
341    
342      def test_resetCoefficient_InHomogeneousConstraint(self):      def test_resetCoefficient_InHomogeneousConstraint(self):
343          mypde=LinearPDE(self.domain,debug=self.DEBUG)          mypde=LinearPDE(self.domain,debug=self.DEBUG)
344            mypde.setSymmetryOn()
345          x=self.domain.getX()          x=self.domain.getX()
346          mypde.setValue(A=kronecker(self.domain),D=1.,Y=1.,r=1,q=x[0].whereZero())          mypde.setValue(A=kronecker(self.domain),D=1.,Y=1.,r=1,q=whereZero(x[0]))
347          u1=mypde.getSolution()          u1=mypde.getSolution(verbose=self.VERBOSE)
348          mypde.setValue(Y=2.,D=2)          mypde.setValue(Y=2.,D=2)
349          u2=mypde.getSolution()          u2=mypde.getSolution(verbose=self.VERBOSE)
350          self.failUnless(self.check(u2,u1),'solution is wrong.')          self.failUnless(self.check(u2,u1),'first solution is wrong.')
351            u2=mypde.getSolution(verbose=self.VERBOSE)
352            self.failUnless(self.check(u2,u1),'first solution is wrong.')
353          mypde.setValue(r=2,Y=4.)          mypde.setValue(r=2,Y=4.)
354          u2=mypde.getSolution()          u2=mypde.getSolution(verbose=self.VERBOSE)
355          self.failUnless(self.check(u2,2*u1),'solution is wrong.')          self.failUnless(self.check(u2,2*u1),'second solution is wrong.')
356    
357      def test_symmetryCheckTrue_System(self):      def test_symmetryCheckTrue_System(self):
358          d=self.domain.getDim()          d=self.domain.getDim()
# Line 457  class Test_LinearPDE(Test_linearPDEs): Line 460  class Test_LinearPDE(Test_linearPDEs):
460          mypde=LinearPDE(self.domain,debug=self.DEBUG)          mypde=LinearPDE(self.domain,debug=self.DEBUG)
461          mypde.setValue(A=kronecker(self.domain),D=1.,Y=1.)          mypde.setValue(A=kronecker(self.domain),D=1.,Y=1.)
462          mypde.setSolverMethod(mypde.DIRECT)          mypde.setSolverMethod(mypde.DIRECT)
463          u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.ILU0)          u=mypde.getSolution(verbose=self.VERBOSE)
464          self.failUnless(self.check(u,1.),'solution is wrong.')          self.failUnless(self.check(u,1.),'solution is wrong.')
465      def test_BICGSTAB_JACOBI(self):      def test_BICGSTAB_JACOBI(self):
466          mypde=LinearPDE(self.domain,debug=self.DEBUG)          mypde=LinearPDE(self.domain,debug=self.DEBUG)
# Line 487  class Test_LinearPDE(Test_linearPDEs): Line 490  class Test_LinearPDE(Test_linearPDEs):
490          mypde=LinearPDE(self.domain,debug=self.DEBUG)          mypde=LinearPDE(self.domain,debug=self.DEBUG)
491          mypde.setValue(A=kronecker(self.domain),D=1.,Y=1.)          mypde.setValue(A=kronecker(self.domain),D=1.,Y=1.)
492      mypde.setSolverMethod(mypde.GMRES)      mypde.setSolverMethod(mypde.GMRES)
493            # u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.JACOBI,truncation=5)
494          u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.JACOBI)          u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.JACOBI)
495          self.failUnless(self.check(u,1.),'solution is wrong.')          self.failUnless(self.check(u,1.),'solution is wrong.')
496      def test_GMRESnoRestart_ILU0(self):      def test_GMRESnoRestart_ILU0(self):
497          mypde=LinearPDE(self.domain,debug=self.DEBUG)          mypde=LinearPDE(self.domain,debug=self.DEBUG)
498          mypde.setValue(A=kronecker(self.domain),D=1.,Y=1.)          mypde.setValue(A=kronecker(self.domain),D=1.,Y=1.)
499      mypde.setSolverMethod(mypde.GMRES)      mypde.setSolverMethod(mypde.GMRES)
500            # u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.ILU0,truncation=5)
501          u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.ILU0)          u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.ILU0)
502          self.failUnless(self.check(u,1.),'solution is wrong.')          self.failUnless(self.check(u,1.),'solution is wrong.')
503      def test_GMRES_JACOBI(self):      def test_GMRES_JACOBI(self):
# Line 517  class Test_LinearPDE(Test_linearPDEs): Line 522  class Test_LinearPDE(Test_linearPDEs):
522          mypde=LinearPDE(self.domain,debug=self.DEBUG)          mypde=LinearPDE(self.domain,debug=self.DEBUG)
523          mypde.setValue(A=kronecker(self.domain),D=1.,Y=1.)          mypde.setValue(A=kronecker(self.domain),D=1.,Y=1.)
524      mypde.setSolverMethod(mypde.GMRES)      mypde.setSolverMethod(mypde.GMRES)
525          mypde.setTolerance(1.e-12)          u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.ILU0,truncation=10,restart=20)
         u=mypde.getSolution(verbose=self.VERBOSE or True,preconditioner=mypde.ILU0,truncation=10,restart=20)  
526          self.failUnless(self.check(u,1.),'solution is wrong.')          self.failUnless(self.check(u,1.),'solution is wrong.')
527      def test_Lumping_attemptToSetA(self):      def test_Lumping_attemptToSetA(self):
528          mypde=LinearPDE(self.domain,debug=self.DEBUG)          mypde=LinearPDE(self.domain,debug=self.DEBUG)
# Line 561  class Test_LinearPDE(Test_linearPDEs): Line 565  class Test_LinearPDE(Test_linearPDEs):
565          x=self.domain.getX()          x=self.domain.getX()
566          mypde=LinearPDE(self.domain,debug=self.DEBUG)          mypde=LinearPDE(self.domain,debug=self.DEBUG)
567      mypde.setSolverMethod(mypde.LUMPING)      mypde.setSolverMethod(mypde.LUMPING)
568          mypde.setValue(D=1.,Y=1.,q=x[0].whereZero(),r=1.)          mypde.setValue(D=1.,Y=1.,q=whereZero(x[0]),r=1.)
569          u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.ILU0)          u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.ILU0)
570          self.failUnless(self.check(u,1.),'solution is wrong.')          self.failUnless(self.check(u,1.),'solution is wrong.')
571      def test_Lumping_updateRHS(self):      def test_Lumping_updateRHS(self):
# Line 571  class Test_LinearPDE(Test_linearPDEs): Line 575  class Test_LinearPDE(Test_linearPDEs):
575          mypde.setValue(D=1.,Y=1.)          mypde.setValue(D=1.,Y=1.)
576          u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.ILU0)          u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.ILU0)
577          self.failUnless(self.check(u,1.),'first solution is wrong.')          self.failUnless(self.check(u,1.),'first solution is wrong.')
578          mypde.setValue(Y=2.,q=x[0].whereZero(),r=2.)          mypde.setValue(Y=2.,q=whereZero(x[0]),r=2.)
579          u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.ILU0)          u=mypde.getSolution(verbose=self.VERBOSE,preconditioner=mypde.ILU0)
580          self.failUnless(self.check(u,2.),'second solution is wrong.')          self.failUnless(self.check(u,2.),'second solution is wrong.')
581      def test_Lumping_updateOperator(self):      def test_Lumping_updateOperator(self):

Legend:
Removed from v.153  
changed lines
  Added in v.304

  ViewVC Help
Powered by ViewVC 1.1.26