/[escript]/trunk/escript/test/python/test_linearPDEs.py
ViewVC logotype

Annotation of /trunk/escript/test/python/test_linearPDEs.py

Parent Directory Parent Directory | Revision Log Revision Log


Revision 2480 - (hide annotations)
Fri Jun 19 03:35:14 2009 UTC (10 years, 4 months ago) by gross
File MIME type: text/x-python
File size: 171885 byte(s)
test fixed.
1 ksteube 1809
2     ########################################################
3 ksteube 1312 #
4 ksteube 1809 # Copyright (c) 2003-2008 by University of Queensland
5     # Earth Systems Science Computational Center (ESSCC)
6     # http://www.uq.edu.au/esscc
7 ksteube 1312 #
8 ksteube 1809 # Primary Business: Queensland, Australia
9     # Licensed under the Open Software License version 3.0
10     # http://www.opensource.org/licenses/osl-3.0.php
11 ksteube 1312 #
12 ksteube 1809 ########################################################
13 jgs 149
14 ksteube 1809 __copyright__="""Copyright (c) 2003-2008 by University of Queensland
15     Earth Systems Science Computational Center (ESSCC)
16     http://www.uq.edu.au/esscc
17     Primary Business: Queensland, Australia"""
18     __license__="""Licensed under the Open Software License version 3.0
19     http://www.opensource.org/licenses/osl-3.0.php"""
20 jfenwick 2344 __url__="https://launchpad.net/escript-finley"
21 ksteube 1809
22 jgs 149 """
23     Test suite for linearPDEs class
24    
25     The tests must be linked with a Domain class object in the setUp method:
26    
27     from esys.finley import Rectangle
28     class Test_LinearPDEOnFinley(Test_LinearPDE):
29     def setUp(self):
30     self.domain = Rectangle(10,10,2)
31 gross 798 def tearDown(self):
32     del self.domain
33 jgs 149 suite = unittest.TestSuite()
34     suite.addTest(unittest.makeSuite(Test_LinearPDEOnFinley))
35     unittest.TextTestRunner(verbosity=2).run(suite)
36    
37 gross 637 @var __author__: name of author
38     @var __copyright__: copyrights
39     @var __license__: licence agreement
40     @var __url__: url entry point on documentation
41     @var __version__: version
42     @var __date__: date of the version
43 jgs 149 """
44    
45     __author__="Lutz Gross, l.gross@uq.edu.au"
46    
47 gross 2325 from esys.escript.util import Lsup,kronecker,interpolate,whereZero, outer, swap_axes
48 ksteube 1312 from esys.escript import Function,FunctionOnBoundary,FunctionOnContactZero,Solution,ReducedSolution,Vector,ContinuousFunction,Scalar, ReducedFunction,ReducedFunctionOnBoundary,ReducedFunctionOnContactZero,Data, Tensor4, Tensor
49 gross 2470 from esys.escript.linearPDEs import LinearPDE,IllegalCoefficientValue,Poisson, IllegalCoefficientFunctionSpace, TransportPDE, IllegalCoefficient, Helmholtz, LameEquation, SolverOptions
50 jfenwick 2455 import numpy
51 jgs 149 import unittest
52    
53     class Test_linearPDEs(unittest.TestCase):
54     TOL=1.e-6
55 jgs 153 SOLVER_TOL=1.e-10
56 jgs 149 DEBUG=False
57     VERBOSE=False
58     def check(self,arg,ref_arg,tol=None):
59     """
60     checks if arg and ref_arg are nearly identical using the L{Lsup<esys.escript.util.Lsup>}
61     """
62     if tol==None: tol=self.TOL
63     return Lsup(arg-ref_arg)<=tol*Lsup(ref_arg)
64    
65 gross 2325 class Test_LameEquation(Test_linearPDEs):
66    
67     def test_config(self):
68     mypde=LameEquation(self.domain,debug=self.DEBUG)
69     d=self.domain.getDim()
70 gross 2474 self.failUnlessEqual((mypde.getNumEquations(), mypde.getNumSolutions(), mypde.getSolverOptions().isSymmetric()),(d,d,True),"set up incorrect")
71 gross 2325
72     def test_setCoefficient_q(self):
73     mypde=LameEquation(self.domain,debug=self.DEBUG)
74     x=self.domain.getX()
75     mypde.setValue(q=x)
76    
77     q_ref=interpolate(x,Solution(self.domain))
78     self.failUnless(self.check(mypde.getCoefficient("A"),0),"A is not 0")
79     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
80     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
81     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
82     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
83     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
84     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
85     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
86     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
87     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
88     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
89     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
90     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
91     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
92     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
93     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"Y_reduced is not empty")
94     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
95     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
96     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
97     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
98     self.failUnless(self.check(mypde.getCoefficient("q"),q_ref),"q is not empty")
99     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
100    
101     def test_setCoefficient_r(self):
102     mypde=LameEquation(self.domain,debug=self.DEBUG)
103     x=self.domain.getX()
104     mypde.setValue(r=x)
105    
106     r_ref=interpolate(x,Solution(self.domain))
107     self.failUnless(self.check(mypde.getCoefficient("A"),0),"A is not 0")
108     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
109     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
110     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
111     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
112     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
113     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
114     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
115     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
116     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
117     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
118     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
119     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
120     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
121     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
122     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"Y_reduced is not empty")
123     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
124     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
125     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
126     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
127     self.failUnless(self.check(mypde.getCoefficient("r"),r_ref),"r is nor x")
128     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
129    
130    
131     def test_setCoefficient_F(self):
132     mypde=LameEquation(self.domain,debug=self.DEBUG)
133     x=self.domain.getX()
134     mypde.setValue(F=x)
135    
136     Y_ref=interpolate(x,Function(self.domain))
137     self.failUnless(self.check(mypde.getCoefficient("A"),0),"A is not 0")
138     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
139     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
140     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
141     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
142     self.failUnless(self.check(mypde.getCoefficient("Y"),Y_ref),"Y is not x")
143     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
144     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
145     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
146     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
147     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
148     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
149     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
150     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
151     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
152     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"Y_reduced is not empty")
153     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
154     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
155     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
156     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
157     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
158     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
159    
160     def test_setCoefficient_f(self):
161     mypde=LameEquation(self.domain,debug=self.DEBUG)
162     x=self.domain.getX()
163     mypde.setValue(f=x)
164    
165     y_ref=interpolate(x,FunctionOnBoundary(self.domain))
166     self.failUnless(self.check(mypde.getCoefficient("A"),0),"A is not 0")
167     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
168     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
169     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
170     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
171     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
172     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
173     self.failUnless(self.check(mypde.getCoefficient("y"),y_ref),"d is not x[0]")
174     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
175     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
176     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
177     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
178     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
179     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
180     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
181     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"X_reduced is not empty")
182     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
183     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
184     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
185     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
186     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
187     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
188    
189     def test_setCoefficient_sigma(self):
190     mypde=LameEquation(self.domain,debug=self.DEBUG)
191     x=self.domain.getX()
192     mypde.setValue(sigma=outer(x,x))
193    
194     X_ref=interpolate(outer(x,x),Function(self.domain))
195     self.failUnless(self.check(mypde.getCoefficient("A"),0),"A is not 0")
196     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
197     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
198     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
199     self.failUnless(self.check(mypde.getCoefficient("X"),X_ref),"X is not x X x")
200     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
201     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
202     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
203     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
204     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
205     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
206     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
207     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
208     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
209     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
210     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"X_reduced is not empty")
211     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
212     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
213     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
214     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
215     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
216     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
217    
218     def test_setCoefficient_lambda(self):
219     mypde=LameEquation(self.domain,debug=self.DEBUG)
220     x=self.domain.getX()
221     mypde.setValue(lame_lambda=x[0])
222    
223    
224     k3=kronecker(Function(self.domain))
225     k3Xk3=outer(k3,k3)
226     A_ref=x[0]*k3Xk3
227    
228     self.failUnless(self.check(mypde.getCoefficient("A"),A_ref),"A is not kronecker")
229     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
230     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
231     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
232     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
233     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
234     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
235     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
236     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
237     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
238     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
239     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
240     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
241     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
242     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
243     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"Y_reduced is not empty")
244     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
245     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
246     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
247     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
248     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
249     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
250    
251     def test_setCoefficient_mu(self):
252     mypde=LameEquation(self.domain,debug=self.DEBUG)
253     x=self.domain.getX()
254     mypde.setValue(lame_mu=x[0])
255    
256    
257     k3=kronecker(Function(self.domain))
258     k3Xk3=outer(k3,k3)
259     A_ref=x[0]*(swap_axes(k3Xk3,0,3)+swap_axes(k3Xk3,1,3))
260    
261     self.failUnless(self.check(mypde.getCoefficient("A"),A_ref),"A is not kronecker")
262     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
263     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
264     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
265     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
266     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
267     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
268     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
269     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
270     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
271     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
272     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
273     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
274     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
275     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
276     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"Y_reduced is not empty")
277     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
278     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
279     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
280     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
281     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
282     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
283    
284     def test_setCoefficient_lambdamu(self):
285     mypde=LameEquation(self.domain,debug=self.DEBUG)
286     x=self.domain.getX()
287     mypde.setValue(lame_lambda=x[0], lame_mu=x[1])
288    
289     k3=kronecker(Function(self.domain))
290     k3Xk3=outer(k3,k3)
291     A_ref=x[0]*k3Xk3+x[1]*(swap_axes(k3Xk3,0,3)+swap_axes(k3Xk3,1,3))
292    
293     self.failUnless(self.check(mypde.getCoefficient("A"),A_ref),"A is not kronecker")
294     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
295     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
296     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
297     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
298     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
299     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
300     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
301     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
302     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
303     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
304     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
305     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
306     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
307     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
308     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"Y_reduced is not empty")
309     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
310     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
311     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
312     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
313     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
314     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
315    
316     def test_solve(self):
317     d=self.domain.getDim()
318     mypde=LameEquation(self.domain)
319     cf=ContinuousFunction(self.domain)
320     x=cf.getX()
321     u_ex=x
322     msk=Vector(0.,cf)
323     for i in range(d): msk[i]=whereZero(x[i])
324     mypde.setValue(q=msk,r=u_ex,lame_mu=3,lame_lambda=50,f=(2*3+50*d)*FunctionOnBoundary(self.domain).getNormal())
325    
326     u=mypde.getSolution()
327     self.failUnless(self.check(u,u_ex,10*self.TOL),"incorrect solution")
328    
329 gross 2323 class Test_Helmholtz(Test_linearPDEs):
330    
331     def test_config(self):
332     mypde=Helmholtz(self.domain,debug=self.DEBUG)
333 gross 2474 self.failUnlessEqual((mypde.getNumEquations(), mypde.getNumSolutions(), mypde.getSolverOptions().isSymmetric()),(1,1,True),"set up incorrect")
334 gross 2323 def test_setCoefficient_q(self):
335     mypde=Helmholtz(self.domain,debug=self.DEBUG)
336     x=self.domain.getX()
337     mypde.setValue(q=whereZero(x[0]))
338    
339     q_ref=interpolate(whereZero(x[0]),Solution(self.domain))
340     A_ref=kronecker(self.domain)
341    
342     self.failUnless(self.check(mypde.getCoefficient("A"),A_ref),"A is not kronecker")
343     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
344     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
345     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
346     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
347     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
348     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
349     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
350     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
351     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
352     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
353     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
354     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
355     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
356     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
357     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"Y_reduced is not empty")
358     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
359     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
360     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
361     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
362     self.failUnless(self.check(mypde.getCoefficient("q"),q_ref),"q is not empty")
363     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
364    
365     def test_setCoefficient_r(self):
366     mypde=Helmholtz(self.domain,debug=self.DEBUG)
367     x=self.domain.getX()
368     mypde.setValue(r=x[0])
369    
370     r_ref=interpolate(x[0],Solution(self.domain))
371     A_ref=kronecker(self.domain)
372     self.failUnless(self.check(mypde.getCoefficient("A"),A_ref),"A is not kronecker")
373     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
374     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
375     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
376     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
377     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
378     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
379     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
380     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
381     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
382     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
383     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
384     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
385     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
386     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
387     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"Y_reduced is not empty")
388     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
389     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
390     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
391     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
392     self.failUnless(self.check(mypde.getCoefficient("r"),r_ref),"r is nor x[0]")
393     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
394    
395    
396     def test_setCoefficient_f(self):
397     mypde=Helmholtz(self.domain,debug=self.DEBUG)
398     x=self.domain.getX()
399     mypde.setValue(f=x[0])
400    
401     Y_ref=interpolate(x[0],Function(self.domain))
402     A_ref=kronecker(self.domain)
403     self.failUnless(self.check(mypde.getCoefficient("A"),A_ref),"A is not kronecker")
404     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
405     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
406     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
407     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
408     self.failUnless(self.check(mypde.getCoefficient("Y"),Y_ref),"Y is not x[0]")
409     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
410     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
411     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
412     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
413     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
414     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
415     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
416     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
417     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
418     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"Y_reduced is not empty")
419     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
420     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
421     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
422     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
423     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
424     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
425    
426     def test_setCoefficient_alpha(self):
427     mypde=Helmholtz(self.domain,debug=self.DEBUG)
428     x=self.domain.getX()
429     mypde.setValue(alpha=x[0])
430    
431     d_ref=interpolate(x[0],FunctionOnBoundary(self.domain))
432     A_ref=kronecker(self.domain)
433     self.failUnless(self.check(mypde.getCoefficient("A"),A_ref),"A is not kronecker")
434     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
435     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
436     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
437     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
438     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
439     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
440     self.failUnless(self.check(mypde.getCoefficient("d"),d_ref),"d is not x[0]")
441     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
442     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
443     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
444     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
445     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
446     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
447     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
448     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"X_reduced is not empty")
449     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
450     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
451     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
452     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
453     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
454     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
455    
456     def test_setCoefficient_g(self):
457     mypde=Helmholtz(self.domain,debug=self.DEBUG)
458     x=self.domain.getX()
459     mypde.setValue(g=x[0])
460    
461     y_ref=interpolate(x[0],FunctionOnBoundary(self.domain))
462     A_ref=kronecker(self.domain)
463     self.failUnless(self.check(mypde.getCoefficient("A"),A_ref),"A is not kronecker")
464     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
465     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
466     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
467     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
468     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
469     self.failUnless(self.check(mypde.getCoefficient("y"),y_ref),"y is not x[0]")
470     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
471     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
472     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
473     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
474     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
475     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
476     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
477     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
478     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"Y_reduced is not empty")
479     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
480     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
481     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
482     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
483     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
484     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
485    
486     def test_setCoefficient_omega(self):
487     mypde=Helmholtz(self.domain,debug=self.DEBUG)
488     x=self.domain.getX()
489     mypde.setValue(omega=x[0])
490    
491     D_ref=interpolate(x[0],Function(self.domain))
492     A_ref=kronecker(self.domain)
493     self.failUnless(self.check(mypde.getCoefficient("A"),A_ref),"A is not kronecker")
494     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
495     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
496     self.failUnless(self.check(mypde.getCoefficient("D"),D_ref),"D is not x[0]")
497     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
498     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
499     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
500     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
501     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
502     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
503     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
504     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
505     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
506     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
507     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
508     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"Y_reduced is not empty")
509     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
510     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
511     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
512     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
513     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
514     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
515    
516     def test_solve(self):
517     d=self.domain.getDim()
518     cf=ContinuousFunction(self.domain)
519     u_ex=Scalar(1.,cf)
520     mypde=Helmholtz(self.domain)
521     mypde.setValue(f=3,omega=3,alpha=2,g=2)
522     u=mypde.getSolution()
523     self.failUnless(self.check(u,u_ex,10*self.TOL),"incorrect solution")
524    
525 jgs 149 class Test_Poisson(Test_linearPDEs):
526    
527     def test_config(self):
528     mypde=Poisson(self.domain,debug=self.DEBUG)
529 gross 2474 self.failUnlessEqual((mypde.getNumEquations(), mypde.getNumSolutions(), mypde.getSolverOptions().isSymmetric()),(1,1,True),"set up incorrect")
530 jgs 149 def test_setCoefficient_q(self):
531     mypde=Poisson(self.domain,debug=self.DEBUG)
532     x=self.domain.getX()
533 gross 304 q_ref=interpolate(whereZero(x[0]),Solution(self.domain))
534 jgs 149 A_ref=kronecker(self.domain)
535 gross 304 mypde.setValue(q=whereZero(x[0]))
536 gross 1841 self.failUnless(self.check(mypde.getCoefficient("A"),A_ref),"A is not kronecker")
537     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
538     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
539     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
540     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
541     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
542     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
543     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
544     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
545     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
546     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
547     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
548     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
549     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
550     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
551     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"Y_reduced is not empty")
552     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
553     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
554     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
555     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
556     self.failUnless(self.check(mypde.getCoefficient("q"),q_ref),"q is not empty")
557     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
558 jgs 149 def test_setCoefficient_f(self):
559     mypde=Poisson(self.domain,debug=self.DEBUG)
560     x=self.domain.getX()
561     Y_ref=interpolate(x[0],Function(self.domain))
562     A_ref=kronecker(self.domain)
563     mypde.setValue(f=x[0])
564 gross 1841 self.failUnless(self.check(mypde.getCoefficient("A"),A_ref),"A is not kronecker")
565     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
566     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
567     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
568     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
569     self.failUnless(self.check(mypde.getCoefficient("Y"),Y_ref),"Y is not x[0]")
570     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
571     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
572     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
573     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
574     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
575     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
576     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
577     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
578     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
579     self.failUnless(mypde.getCoefficient("Y_reduced").isEmpty(),"Y_reduced is not empty")
580     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
581     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
582     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
583     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
584     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
585     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
586 gross 1072 def test_setCoefficient_f_reduced(self):
587     mypde=Poisson(self.domain,debug=self.DEBUG)
588     x=self.domain.getX()
589     Y_ref=interpolate(x[0],ReducedFunction(self.domain))
590     A_ref=kronecker(self.domain)
591     mypde.setValue(f_reduced=x[0])
592 gross 1841 self.failUnless(self.check(mypde.getCoefficient("A"),A_ref),"A is not kronecker")
593     self.failUnless(mypde.getCoefficient("B").isEmpty(),"B is not empty")
594     self.failUnless(mypde.getCoefficient("C").isEmpty(),"C is not empty")
595     self.failUnless(mypde.getCoefficient("D").isEmpty(),"D is not empty")
596     self.failUnless(mypde.getCoefficient("X").isEmpty(),"X is not empty")
597     self.failUnless(mypde.getCoefficient("Y").isEmpty(),"Y is not empty")
598     self.failUnless(mypde.getCoefficient("y").isEmpty(),"y is not empty")
599     self.failUnless(mypde.getCoefficient("d").isEmpty(),"d is not empty")
600     self.failUnless(mypde.getCoefficient("d_contact").isEmpty(),"d_contact is not empty")
601     self.failUnless(mypde.getCoefficient("y_contact").isEmpty(),"y_contact is not empty")
602     self.failUnless(mypde.getCoefficient("A_reduced").isEmpty(),"A_reduced is not empty")
603     self.failUnless(mypde.getCoefficient("B_reduced").isEmpty(),"B_reduced is not empty")
604     self.failUnless(mypde.getCoefficient("C_reduced").isEmpty(),"C_reduced is not empty")
605     self.failUnless(mypde.getCoefficient("D_reduced").isEmpty(),"D_reduced is not empty")
606     self.failUnless(mypde.getCoefficient("X_reduced").isEmpty(),"X_reduced is not empty")
607     self.failUnless(self.check(mypde.getCoefficient("Y_reduced"),Y_ref),"Y_reduced is not x[0]")
608     self.failUnless(mypde.getCoefficient("y_reduced").isEmpty(),"y_reduced is not empty")
609     self.failUnless(mypde.getCoefficient("d_reduced").isEmpty(),"d_reduced is not empty")
610     self.failUnless(mypde.getCoefficient("d_contact_reduced").isEmpty(),"d_contact_reduced is not empty")
611     self.failUnless(mypde.getCoefficient("y_contact_reduced").isEmpty(),"y_contact_reduced is not empty")
612     self.failUnless(mypde.getCoefficient("q").isEmpty(),"q is not empty")
613     self.failUnless(mypde.getCoefficient("r").isEmpty(),"r is not empty")
614 jgs 149 def test_solve(self):
615     d=self.domain.getDim()
616     cf=ContinuousFunction(self.domain)
617     x=cf.getX()
618     #construct exact solution:
619     u_ex=Scalar(1.,cf)
620     for i in range(d):
621     u_ex*=x[i]*(2.-x[i])
622     #construct mask:
623     msk=Scalar(0.,cf)
624     for i in range(d):
625 gross 304 msk+=whereZero(x[i])
626 jgs 149 #construct right hand side
627     f=Scalar(0,cf)
628     for i in range(d):
629     f_p=Scalar(1,cf)
630     for j in range(d):
631     if i==j:
632     f_p*=2.
633     else:
634     f_p*=x[j]*(2-x[j])
635     f+=f_p
636     mypde=Poisson(self.domain)
637     mypde.setValue(f=f,q=msk)
638     u=mypde.getSolution()
639     self.failUnless(self.check(u,u_ex,10*self.TOL),"incorrect solution")
640    
641 gross 855 class Test_LinearPDE_noLumping(Test_linearPDEs):
642 jgs 149 N=4
643 gross 2470 def test_SolverOptions(self):
644     so=SolverOptions()
645    
646     self.failUnless(so.getLevelMax() == 10, "initial LevelMax is wrong.")
647     self.failUnlessRaises(ValueError,so.setLevelMax,-1)
648     so.setLevelMax(3)
649     self.failUnless(so.getLevelMax() == 3, "LevelMax is wrong.")
650    
651     self.failUnless(so.getCoarseningThreshold() == 0.05, "initial CoarseningThreshold is wrong.")
652     self.failUnlessRaises(ValueError,so.setCoarseningThreshold,-1)
653     so.setCoarseningThreshold(0.1)
654     self.failUnless(so.getCoarseningThreshold() == 0.1, "CoarseningThreshold is wrong.")
655    
656     self.failUnless(so.getNumSweeps() == 2, "initial Sweeps is wrong.")
657     self.failUnlessRaises(ValueError,so.setNumSweeps,-1)
658     so.setNumSweeps(3)
659     self.failUnless(so.getNumSweeps() == 3, "Sweeps is wrong.")
660    
661     self.failUnless(so.getNumPreSweeps() == 2, "initial PreSweeps is wrong.")
662     self.failUnlessRaises(ValueError,so.setNumPreSweeps,-1)
663     so.setNumPreSweeps(4)
664     self.failUnless(so.getNumPreSweeps() == 4, "PreSweeps is wrong.")
665    
666     self.failUnless(so.getNumPostSweeps() == 2, "initial PreSweeps is wrong.")
667     self.failUnlessRaises(ValueError,so.setNumPostSweeps,-1)
668     so.setNumPostSweeps(5)
669     self.failUnless(so.getNumPostSweeps() == 5, "PreSweeps is wrong.")
670    
671     self.failUnless(so.getTolerance() == 1.e-8, "initial Tolerance is wrong.")
672     self.failUnlessRaises(ValueError,so.setTolerance,-1)
673     so.setTolerance(0.2)
674     self.failUnless(so.getTolerance() == 0.2, "Tolerance is wrong.")
675    
676     self.failUnless(so.getAbsoluteTolerance() == 0., "initial AbsoluteTolerance is wrong.")
677     self.failUnlessRaises(ValueError,so.setAbsoluteTolerance,-1)
678     so.setAbsoluteTolerance(0.3)
679     self.failUnless(so.getAbsoluteTolerance() == 0.3, "AbsoluteTolerance is wrong.")
680    
681     self.failUnless(so.getInnerTolerance() == 0.9, "initial InnerTolerance is wrong.")
682     self.failUnlessRaises(ValueError,so.setInnerTolerance,-1)
683     so.setInnerTolerance(0.4)
684     self.failUnless(so.getInnerTolerance() == 0.4, "InnerTolerance is wrong.")
685    
686     self.failUnless(so.getDropTolerance() == 0.01, "initial DropTolerance is wrong.")
687     self.failUnlessRaises(ValueError,so.setDropTolerance,-1)
688     so.setDropTolerance(0.5)
689     self.failUnless(so.getDropTolerance() == 0.5, "DropDropTolerance is wrong.")
690    
691     self.failUnless(so.getDropStorage() == 2., "initial DropStorage is wrong.")
692     self.failUnlessRaises(ValueError,so.setDropStorage,-1)
693     so.setDropStorage(10)
694     self.failUnless(so.getDropStorage() == 10, "DropStorage is wrong.")
695    
696     self.failUnless(so.getRelaxationFactor() == 0.3, "initial RelaxationFactor is wrong.")
697     self.failUnlessRaises(ValueError,so.setRelaxationFactor,-1)
698     so.setRelaxationFactor(0.1)
699     self.failUnless(so.getRelaxationFactor() == 0.1, "Relaxation is wrong.")
700    
701    
702 gross 2480 self.failUnless(so.getIterMax() == 100000, "initial IterMax is wrong.")
703 gross 2470 self.failUnlessRaises(ValueError,so.setIterMax,0)
704     so.setIterMax(11)
705     self.failUnless(so.getIterMax() == 11, "IterMax is wrong.")
706    
707     self.failUnless(so.getInnerIterMax() == 10, "initial InnerIterMax is wrong.")
708     self.failUnlessRaises(ValueError,so.setInnerIterMax,0)
709     so.setInnerIterMax(12)
710     self.failUnless(so.getInnerIterMax() == 12, "InnerIterMax is wrong.")
711    
712     self.failUnless(so.getTruncation() == 20, "initial Truncation is wrong.")
713     self.failUnlessRaises(ValueError,so.setTruncation,0)
714     so.setTruncation(13)
715     self.failUnless(so.getTruncation() == 13, "Truncation is wrong.")
716    
717     self.failUnless(so.getRestart() == None, "initial Truncation is wrong.")
718     self.failUnlessRaises(ValueError,so.setTruncation,0)
719     so.setRestart(14)
720     self.failUnless(so.getRestart() == 14, "Truncation is wrong.")
721     so.setRestart(None)
722     self.failUnless(so.getRestart() == None, "Truncation is wrong.")
723 gross 2474
724     self.failUnless(not so.isVerbose(), "initial verbosity flag is wrong.")
725     so.setVerbosityOn()
726     self.failUnless(so.isVerbose(), "verbosity (1) flag is wrong.")
727     so.setVerbosityOff()
728     self.failUnless(not so.isVerbose(), "verbosity (2) flag is wrong.")
729     so.setVerbosity(verbose=True)
730     self.failUnless(so.isVerbose(), "verbosity (3) flag is wrong.")
731     so.setVerbosity(verbose=False)
732     self.failUnless(not so.isVerbose(), "verbosity (4) flag is wrong.")
733    
734 gross 2470 self.failUnless(not so.isSymmetric(), "initial symmetry flag is wrong.")
735     so.setSymmetryOn()
736     self.failUnless(so.isSymmetric(), "symmetry (1) flag is wrong.")
737     so.setSymmetryOff()
738     self.failUnless(not so.isSymmetric(), "symmetry (2) flag is wrong.")
739     so.setSymmetry(flag=True)
740     self.failUnless(so.isSymmetric(), "symmetry (3) flag is wrong.")
741     so.setSymmetry(flag=False)
742     self.failUnless(not so.isSymmetric(), "symmetry (4) flag is wrong.")
743    
744     self.failUnless(so.adaptInnerTolerance(), "initial InnerToleranceAdaption flag is wrong.")
745     so.setInnerToleranceAdaptionOn()
746     self.failUnless(so.adaptInnerTolerance(), "InnerToleranceAdaption (1) flag is wrong.")
747     so.setInnerToleranceAdaptionOff()
748     self.failUnless(not so.adaptInnerTolerance(), "InnerToleranceAdaption (2) flag is wrong.")
749     so.setInnerToleranceAdaption(adapt=True)
750     self.failUnless(so.adaptInnerTolerance(), "InnerToleranceAdaption (3) flag is wrong.")
751     so.setInnerToleranceAdaption(adapt=False)
752     self.failUnless(not so.adaptInnerTolerance(), "InnerToleranceAdaption (4) flag is wrong.")
753    
754     self.failUnless(not so.acceptConvergenceFailure(), "initial acceptConvergenceFailure flag is wrong.")
755     so.setAcceptanceConvergenceFailureOn()
756     self.failUnless(so.acceptConvergenceFailure(), "acceptConvergenceFailure (1) flag is wrong.")
757     so.setAcceptanceConvergenceFailureOff()
758     self.failUnless(not so.acceptConvergenceFailure(), "acceptConvergenceFailure (2) flag is wrong.")
759     so.setAcceptanceConvergenceFailure(accept=True)
760     self.failUnless(so.acceptConvergenceFailure(), "acceptConvergenceFailure (3) flag is wrong.")
761     so.setAcceptanceConvergenceFailure(accept=False)
762     self.failUnless(not so.acceptConvergenceFailure(), "acceptConvergenceFailure (4) flag is wrong.")
763    
764     self.failUnless(so.getReordering() == 30, "initial Reordering is wrong.")
765     self.failUnlessRaises(ValueError,so.setReordering,-1)
766     so.setReordering(so.NO_REORDERING)
767     self.failUnless(so.getReordering() == 17, "NO_REORDERING is not set.")
768     so.setReordering(so.MINIMUM_FILL_IN)
769     self.failUnless(so.getReordering() == 18, "MINIMUM_FILL_IN is not set.")
770     so.setReordering(so.NESTED_DISSECTION)
771     self.failUnless(so.getReordering() == 19, "NESTED_DISSECTION is not set.")
772     so.setReordering(so.DEFAULT_REORDERING)
773     self.failUnless(so.getReordering() == 30, "DEFAULT_REORDERING is not set.")
774    
775     self.failUnless(so.getPackage() == 0, "initial solver package is wrong.")
776     self.failUnlessRaises(ValueError,so.setPackage,-1)
777     so.setPackage(so.PASO)
778     self.failUnless(so.getPackage() == 21, "PASO is not set.")
779     so.setPackage(so.SUPER_LU)
780     self.failUnless(so.getPackage() == 31, "SUPER_LU is not set.")
781     so.setPackage(so.PASTIX)
782     self.failUnless(so.getPackage() == 32, "PASTIX is not set.")
783     so.setPackage(so.MKL)
784     self.failUnless(so.getPackage() == 15, "MKL is not set.")
785     so.setPackage(so.UMFPACK)
786     self.failUnless(so.getPackage() == 16, "UMFPACK is not set.")
787     so.setPackage(so.TRILINOS)
788     self.failUnless(so.getPackage() == 24, "TRILINOS is not set.")
789    
790     self.failUnless(so.getSolverMethod() == 0, "initial SolverMethod is wrong.")
791     self.failUnlessRaises(ValueError,so.setSolverMethod,-1)
792     so.setSolverMethod(so.DIRECT)
793     self.failUnless(so.getSolverMethod() == 1, "DIRECT is not set.")
794     so.setSolverMethod(so.CHOLEVSKY)
795     self.failUnless(so.getSolverMethod() == 2, "CHOLEVSKY is not set.")
796     so.setSolverMethod(so.PCG)
797     self.failUnless(so.getSolverMethod() == 3, "PCG is not set.")
798     so.setSolverMethod(so.CR)
799     self.failUnless(so.getSolverMethod() == 4, "CR is not set.")
800     so.setSolverMethod(so.CGS)
801     self.failUnless(so.getSolverMethod() == 5, "CGS is not set.")
802     so.setSolverMethod(so.BICGSTAB)
803     self.failUnless(so.getSolverMethod() == 6, "BICGSTAB is not set.")
804     so.setSolverMethod(so.SSOR)
805     self.failUnless(so.getSolverMethod() == 7, "SSOR is not set.")
806     so.setSolverMethod(so.GMRES)
807     self.failUnless(so.getSolverMethod() == 11, "GMRES is not set.")
808     so.setSolverMethod(so.PRES20)
809     self.failUnless(so.getSolverMethod() == 12, "PRES20 is not set.")
810     so.setSolverMethod(so.LUMPING)
811     self.failUnless(so.getSolverMethod() == 13, "LUMPING is not set.")
812     so.setSolverMethod(so.ITERATIVE)
813     self.failUnless(so.getSolverMethod() == 20, "ITERATIVE is not set.")
814     so.setSolverMethod(so.AMG)
815     self.failUnless(so.getSolverMethod() == 22, "AMG is not set.")
816     so.setSolverMethod(so.NONLINEAR_GMRES)
817     self.failUnless(so.getSolverMethod() == 25, "NONLINEAR_GMRES is not set.")
818     so.setSolverMethod(so.TFQMR)
819     self.failUnless(so.getSolverMethod() == 26, "TFQMR is not set.")
820     so.setSolverMethod(so.MINRES)
821     self.failUnless(so.getSolverMethod() == 27, "MINRES is not set.")
822     so.setSolverMethod(so.GAUSS_SEIDEL)
823     self.failUnless(so.getSolverMethod() == 28, "GAUSS_SEIDEL is not set.")
824     so.setSolverMethod(so.DEFAULT)
825     self.failUnless(so.getSolverMethod() == 0, "DEFAULT is not set.")
826    
827     self.failUnless(so.getPreconditioner() == 10, "initial Preconditioner is wrong.")
828     self.failUnlessRaises(ValueError,so.setPreconditioner,-1)
829     so.setPreconditioner(so.ILU0)
830     self.failUnless(so.getPreconditioner() == 8, "ILU0 is not set.")
831     so.setPreconditioner(so.SSOR)
832     self.failUnless(so.getPreconditioner() == 7, "SSOR is not set.")
833     so.setPreconditioner(so.ILUT)
834     self.failUnless(so.getPreconditioner() == 9, "ILUT is not set.")
835     so.setPreconditioner(so.JACOBI)
836     self.failUnless(so.getPreconditioner() == 10, "JACOBI is not set.")
837     so.setPreconditioner(so.AMG)
838     self.failUnless(so.getPreconditioner() == 22, "AMG is not set.")
839     so.setPreconditioner(so.REC_ILU)
840     self.failUnless(so.getPreconditioner() == 23, "REC_ILU is not set.")
841     so.setPreconditioner(so.GAUSS_SEIDEL)
842     self.failUnless(so.getPreconditioner() == 28, "GAUSS_SEIDEL is not set.")
843     so.setPreconditioner(so.RILU)
844     self.failUnless(so.getPreconditioner() == 29, "RILU is not set.")
845     so.setPreconditioner(so.NO_PRECONDITIONER)
846     self.failUnless(so.getPreconditioner() == 36, "NO_PRECONDITIONER is not set.")
847    
848     self.failUnless(so.getCoarsening() == 0, "initial Coarseningr is wrong.")
849     self.failUnlessRaises(ValueError,so.setCoarsening,-1)
850     so.setCoarsening(so.YAIR_SHAPIRA_COARSENING)
851     self.failUnless(so.getCoarsening() == 33, "YAIR_SHAPIRA_COARSENING is not set.")
852     so.setCoarsening(so.RUGE_STUEBEN_COARSENING)
853     self.failUnless(so.getCoarsening() == 34, "RUGE_STUEBEN_COARSENING is not set.")
854     so.setCoarsening(so.AGGREGATION_COARSENING)
855     self.failUnless(so.getCoarsening() == 35, "AGREGATION_COARSENING is not set.")
856     so.setCoarsening(so.DEFAULT)
857     self.failUnless(so.getCoarsening() == 0, "DEFAULT is not set.")
858    
859     self.failUnless(so.getDiagnostics("num_iter") == None, "initial num_iter is wrong.")
860     self.failUnless(so.getDiagnostics("num_inner_iter") == None, "initial num_inner_iter is wrong.")
861     self.failUnless(so.getDiagnostics("time") == None, "initial time is wrong.")
862     self.failUnless(so.getDiagnostics("set_up_time") == None, "initial set_up_time is wrong.")
863     self.failUnless(so.getDiagnostics("residual_norm") == None, "initial residual_norm is wrong.")
864     self.failUnless(so.getDiagnostics("converged") == None, "initial converged is wrong.")
865     self.failUnless(so.hasConverged() == None, "initial convergence flag is wrong.")
866     self.failUnless(so.getDiagnostics("cum_num_inner_iter") == 0, "initial cum_num_inner_iter is wrong.")
867     self.failUnless(so.getDiagnostics("cum_num_iter") == 0, "initial cum_num_iter is wrong.")
868     self.failUnless(so.getDiagnostics("cum_time") ==0, "initial cum_time is wrong.")
869     self.failUnless(so.getDiagnostics("cum_set_up_time") == 0, "initial cum_set_up_time is wrong.")
870    
871     so._updateDiagnostics("num_iter",1)
872     so._updateDiagnostics("num_inner_iter",2)
873     so._updateDiagnostics("time",3)
874     so._updateDiagnostics("set_up_time",4)
875     so._updateDiagnostics("residual_norm",5)
876     so._updateDiagnostics("converged",True)
877    
878     self.failUnless(so.getDiagnostics("num_iter") == 1, "num_iter is wrong.")
879     self.failUnless(so.getDiagnostics("num_inner_iter") == 2, "num_inner_iter is wrong.")
880     self.failUnless(so.getDiagnostics("time") == 3, "time is wrong.")
881     self.failUnless(so.getDiagnostics("set_up_time") == 4, "set_up_time is wrong.")
882     self.failUnless(so.getDiagnostics("residual_norm") == 5, "residual_norm is wrong.")
883     self.failUnless(so.getDiagnostics("converged"), "converged is wrong.")
884     self.failUnless(so.hasConverged(), "convergence flag is wrong.")
885     self.failUnless(so.getDiagnostics("cum_num_inner_iter") == 2, "cum_num_inner_iter is wrong.")
886     self.failUnless(so.getDiagnostics("cum_num_iter") == 1, "cum_num_iter is wrong.")
887     self.failUnless(so.getDiagnostics("cum_time") ==3, "cum_time is wrong.")
888     self.failUnless(so.getDiagnostics("cum_set_up_time") == 4, "cum_set_up_time is wrong.")
889    
890     so.resetDiagnostics()
891     self.failUnless(so.getDiagnostics("num_iter") == None, "initial num_iter is wrong.")
892     self.failUnless(so.getDiagnostics("num_inner_iter") == None, "initial num_inner_iter is wrong.")
893     self.failUnless(so.getDiagnostics("time") == None, "initial time is wrong.")
894     self.failUnless(so.getDiagnostics("set_up_time") == None, "initial set_up_time is wrong.")
895     self.failUnless(so.getDiagnostics("residual_norm") == None, "initial residual_norm is wrong.")
896     self.failUnless(so.getDiagnostics("converged") == None, "initial converged is wrong.")
897     self.failUnless(so.hasConverged() == None, "initial convergence flag is wrong")
898     self.failUnless(so.getDiagnostics("cum_num_inner_iter") == 2, "cum_num_inner_iter is wrong.")
899     self.failUnless(so.getDiagnostics("cum_num_iter") == 1, "cum_num_iter is wrong.")
900     self.failUnless(so.getDiagnostics("cum_time") ==3, "cum_time is wrong.")
901     self.failUnless(so.getDiagnostics("cum_set_up_time") == 4, "cum_set_up_time is wrong.")
902    
903     so._updateDiagnostics("num_iter",10)
904     so._updateDiagnostics("num_inner_iter",20)
905     so._updateDiagnostics("time",30)
906     so._updateDiagnostics("set_up_time",40)
907     so._updateDiagnostics("residual_norm",50)
908     so._updateDiagnostics("converged",False)
909    
910     self.failUnless(so.getDiagnostics("num_iter") == 10, "num_iter is wrong.")
911     self.failUnless(so.getDiagnostics("num_inner_iter") == 20, "num_inner_iter is wrong.")
912     self.failUnless(so.getDiagnostics("time") == 30, "time is wrong.")
913     self.failUnless(so.getDiagnostics("set_up_time") == 40, "set_up_time is wrong.")
914     self.failUnless(so.getDiagnostics("residual_norm") == 50, "residual_norm is wrong.")
915     self.failUnless(not so.getDiagnostics("converged"), "converged is wrong.")
916     self.failUnless(not so.hasConverged(), "convergence flag is wrong.")
917     self.failUnless(so.getDiagnostics("cum_num_inner_iter") == 22, "cum_num_inner_iter is wrong.")
918     self.failUnless(so.getDiagnostics("cum_num_iter") == 11, "cum_num_iter is wrong.")
919     self.failUnless(so.getDiagnostics("cum_time") ==33, "cum_time is wrong.")
920     self.failUnless(so.getDiagnostics("cum_set_up_time") == 44, "cum_set_up_time is wrong.")
921    
922     so.resetDiagnostics(all=True)
923     self.failUnless(so.getDiagnostics("num_iter") == None, "initial num_iter is wrong.")
924     self.failUnless(so.getDiagnostics("num_inner_iter") == None, "initial num_inner_iter is wrong.")
925     self.failUnless(so.getDiagnostics("time") == None, "initial time is wrong.")
926     self.failUnless(so.getDiagnostics("set_up_time") == None, "initial set_up_time is wrong.")
927     self.failUnless(so.getDiagnostics("residual_norm") == None, "initial residual_norm is wrong.")
928     self.failUnless(so.getDiagnostics("converged") == None, "initial converged is wrong.")
929     self.failUnless(so.hasConverged() == None, "initial convergence flag is wrong.")
930     self.failUnless(so.getDiagnostics("cum_num_inner_iter") == 0, "initial cum_num_inner_iter is wrong.")
931     self.failUnless(so.getDiagnostics("cum_num_iter") == 0, "initial cum_num_iter is wrong.")
932     self.failUnless(so.getDiagnostics("cum_time") ==0, "initial cum_time is wrong.")
933     self.failUnless(so.getDiagnostics("cum_set_up_time") == 0, "initial cum_set_up_time is wrong.")
934    
935 jgs 149 def test_setCoefficient_WithIllegalFunctionSpace(self):
936     mypde=LinearPDE(self.domain,debug=self.DEBUG)
937 gross 2474 self.failUnlessRaises(IllegalCoefficientFunctionSpace, mypde.setValue, C=Vector(0.,FunctionOnBoundary(self.domain)))
938 gross 1859
939     def test_setCoefficient_WithWrongName(self):
940     mypde=LinearPDE(self.domain,debug=self.DEBUG)
941     self.failUnlessRaises(IllegalCoefficient, mypde.setValue, ROMA=0.)
942    
943 jgs 149 def test_resetCoefficient_WithWrongShape(self):
944     mypde=LinearPDE(self.domain,numEquations=2,debug=self.DEBUG)
945 gross 1859 self.failUnlessRaises(IllegalCoefficientValue, mypde.setValue, C=0.)
946    
947 jgs 149 def test_reducedOn(self):
948     mypde=LinearPDE(self.domain,debug=self.DEBUG)
949     x=self.domain.getX()
950     mypde.setReducedOrderOn()
951     mypde.setValue(A=kronecker(self.domain),D=x[0],Y=x[0])
952     u=mypde.getSolution()
953     self.failUnless(self.check(u,1.),'solution is wrong.')
954    
955     def test_attemptToChangeOrderAfterDefinedCoefficient(self):
956     mypde=LinearPDE(self.domain,debug=self.DEBUG)
957     mypde.setValue(D=1.)
958 gross 2474 self.failUnlessRaises(RuntimeError, mypde.setReducedOrderOn)
959 jgs 149
960     def test_reducedOnConfig(self):
961     mypde=LinearPDE(self.domain,debug=self.DEBUG)
962     mypde.setReducedOrderOn()
963 gross 2474 self.failUnlessEqual((mypde.getFunctionSpaceForSolution(), mypde.getFunctionSpaceForEquation()),(ReducedSolution(self.domain),ReducedSolution(self.domain)),"reduced function spaces expected.")
964 jgs 149 #
965     # set coefficients for scalars:
966     #
967     def test_setCoefficient_A_Scalar(self):
968     d=self.domain.getDim()
969     mypde=LinearPDE(self.domain,debug=self.DEBUG)
970 jfenwick 2455 mypde.setValue(A=numpy.ones((d,d)))
971 gross 1841 coeff=mypde.getCoefficient("A")
972 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((d,d),Function(self.domain),1,1))
973 jgs 149 def test_setCoefficient_B_Scalar(self):
974     d=self.domain.getDim()
975     mypde=LinearPDE(self.domain,debug=self.DEBUG)
976 jfenwick 2455 mypde.setValue(B=numpy.ones((d,)))
977 gross 1841 coeff=mypde.getCoefficient("B")
978 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((d,),Function(self.domain),1,1))
979 jgs 149 def test_setCoefficient_C_Scalar(self):
980     d=self.domain.getDim()
981     mypde=LinearPDE(self.domain,debug=self.DEBUG)
982 jfenwick 2455 mypde.setValue(C=numpy.ones((d,)))
983 gross 1841 coeff=mypde.getCoefficient("C")
984 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((d,),Function(self.domain),1,1))
985 jgs 149 def test_setCoefficient_D_Scalar(self):
986     d=self.domain.getDim()
987     mypde=LinearPDE(self.domain,debug=self.DEBUG)
988     mypde.setValue(D=1.)
989 gross 1841 coeff=mypde.getCoefficient("D")
990 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((),Function(self.domain),1,1))
991 jgs 149 def test_setCoefficient_X_Scalar(self):
992     d=self.domain.getDim()
993     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
994 jfenwick 2455 mypde.setValue(X=numpy.ones((d,)))
995 gross 1841 coeff=mypde.getCoefficient("X")
996 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((d,),Function(self.domain),1))
997 jgs 149 def test_setCoefficient_Y_Scalar(self):
998     d=self.domain.getDim()
999     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1000     mypde.setValue(Y=1.)
1001 gross 1841 coeff=mypde.getCoefficient("Y")
1002 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((),Function(self.domain),1))
1003 jgs 149 def test_setCoefficient_y_Scalar(self):
1004     d=self.domain.getDim()
1005     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1006     mypde.setValue(y=1.)
1007 gross 1841 coeff=mypde.getCoefficient("y")
1008 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((),FunctionOnBoundary(self.domain),1))
1009 jgs 149 def test_setCoefficient_d_Scalar(self):
1010     d=self.domain.getDim()
1011     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1012     mypde.setValue(d=1.)
1013 gross 1841 coeff=mypde.getCoefficient("d")
1014 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((),FunctionOnBoundary(self.domain),1,1))
1015 jgs 149 def test_setCoefficient_d_contact_Scalar(self):
1016     d=self.domain.getDim()
1017     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1018     mypde.setValue(d_contact=1.)
1019 gross 1841 coeff=mypde.getCoefficient("d_contact")
1020 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((),FunctionOnContactZero(self.domain),1,1))
1021 jgs 149 def test_setCoefficient_y_contact_Scalar(self):
1022     d=self.domain.getDim()
1023     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1024     mypde.setValue(y_contact=1.)
1025 gross 1841 coeff=mypde.getCoefficient("y_contact")
1026 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((),FunctionOnContactZero(self.domain),1))
1027 gross 1072 def test_setCoefficient_A_reduced_Scalar(self):
1028     d=self.domain.getDim()
1029     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1030 jfenwick 2455 mypde.setValue(A_reduced=numpy.ones((d,d)))
1031 gross 1841 coeff=mypde.getCoefficient("A_reduced")
1032 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((d,d),ReducedFunction(self.domain),1,1))
1033 gross 1072 def test_setCoefficient_B_reduced_Scalar(self):
1034     d=self.domain.getDim()
1035     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1036 jfenwick 2455 mypde.setValue(B_reduced=numpy.ones((d,)))
1037 gross 1841 coeff=mypde.getCoefficient("B_reduced")
1038 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((d,),ReducedFunction(self.domain),1,1))
1039 gross 1072 def test_setCoefficient_C_reduced_Scalar(self):
1040     d=self.domain.getDim()
1041     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1042 jfenwick 2455 mypde.setValue(C_reduced=numpy.ones((d,)))
1043 gross 1841 coeff=mypde.getCoefficient("C_reduced")
1044 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((d,),ReducedFunction(self.domain),1,1))
1045 gross 1072 def test_setCoefficient_D_reduced_Scalar(self):
1046     d=self.domain.getDim()
1047     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1048     mypde.setValue(D_reduced=1.)
1049 gross 1841 coeff=mypde.getCoefficient("D_reduced")
1050 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((),ReducedFunction(self.domain),1,1))
1051 gross 1072 def test_setCoefficient_X_reduced_Scalar(self):
1052     d=self.domain.getDim()
1053     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1054 jfenwick 2455 mypde.setValue(X_reduced=numpy.ones((d,)))
1055 gross 1841 coeff=mypde.getCoefficient("X_reduced")
1056 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((d,),ReducedFunction(self.domain),1))
1057 gross 1072 def test_setCoefficient_Y_reduced_Scalar(self):
1058     d=self.domain.getDim()
1059     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1060     mypde.setValue(Y_reduced=1.)
1061 gross 1841 coeff=mypde.getCoefficient("Y_reduced")
1062 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((),ReducedFunction(self.domain),1))
1063 gross 1072 def test_setCoefficient_y_reduced_Scalar(self):
1064     d=self.domain.getDim()
1065     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1066     mypde.setValue(y_reduced=1.)
1067 gross 1841 coeff=mypde.getCoefficient("y_reduced")
1068 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((),ReducedFunctionOnBoundary(self.domain),1))
1069 gross 1072 def test_setCoefficient_d_reduced_Scalar(self):
1070     d=self.domain.getDim()
1071     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1072     mypde.setValue(d_reduced=1.)
1073 gross 1841 coeff=mypde.getCoefficient("d_reduced")
1074 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((),ReducedFunctionOnBoundary(self.domain),1,1))
1075 gross 1072 def test_setCoefficient_d_contact_reduced_Scalar(self):
1076     d=self.domain.getDim()
1077     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1078     mypde.setValue(d_contact_reduced=1.)
1079 gross 1841 coeff=mypde.getCoefficient("d_contact_reduced")
1080 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((),ReducedFunctionOnContactZero(self.domain),1,1))
1081 gross 1072 def test_setCoefficient_y_contact_reduced_Scalar(self):
1082     d=self.domain.getDim()
1083     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1084     mypde.setValue(y_contact_reduced=1.)
1085 gross 1841 coeff=mypde.getCoefficient("y_contact_reduced")
1086 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((),ReducedFunctionOnContactZero(self.domain),1))
1087 jgs 149 def test_setCoefficient_r_Scalar(self):
1088     d=self.domain.getDim()
1089     mypde=LinearPDE(self.domain,numEquations=3,debug=self.DEBUG)
1090     mypde.setValue(r=1.)
1091 gross 1841 coeff=mypde.getCoefficient("r")
1092 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions()),((),Solution(self.domain),1))
1093 jgs 149 def test_setCoefficient_q_Scalar(self):
1094     d=self.domain.getDim()
1095     mypde=LinearPDE(self.domain,numEquations=3,debug=self.DEBUG)
1096     mypde.setValue(q=1.)
1097 gross 1841 coeff=mypde.getCoefficient("q")
1098 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions()),((),Solution(self.domain),1))
1099 jgs 149 def test_setCoefficient_r_Scalar_reducedOn(self):
1100     d=self.domain.getDim()
1101     mypde=LinearPDE(self.domain,numEquations=3,debug=self.DEBUG)
1102     mypde.setReducedOrderOn()
1103     mypde.setValue(r=1.)
1104 gross 1841 coeff=mypde.getCoefficient("r")
1105 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions()),((),ReducedSolution(self.domain),1))
1106 jgs 149 def test_setCoefficient_q_Scalar_reducedOn(self):
1107     d=self.domain.getDim()
1108     mypde=LinearPDE(self.domain,numEquations=3,debug=self.DEBUG)
1109     mypde.setReducedOrderOn()
1110     mypde.setValue(q=1.)
1111 gross 1841 coeff=mypde.getCoefficient("q")
1112 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions()),((),ReducedSolution(self.domain),1))
1113 jgs 149
1114 gross 1072 def test_setCoefficient_A_reduced_Scalar_usingA(self):
1115     d=self.domain.getDim()
1116     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1117 jfenwick 2455 mypde.setValue(A=Data(numpy.ones((d,d)),ReducedFunction(self.domain)))
1118 gross 1841 coeff=mypde.getCoefficient("A_reduced")
1119 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((d,d),ReducedFunction(self.domain),1,1))
1120 gross 1072 def test_setCoefficient_B_reduced_Scalar_usingB(self):
1121     d=self.domain.getDim()
1122     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1123 jfenwick 2455 mypde.setValue(B=Data(numpy.ones((d,)),ReducedFunction(self.domain)))
1124 gross 1841 coeff=mypde.getCoefficient("B_reduced")
1125 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((d,),ReducedFunction(self.domain),1,1))
1126 gross 1072 def test_setCoefficient_C_reduced_Scalar_usingC(self):
1127     d=self.domain.getDim()
1128     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1129 jfenwick 2455 mypde.setValue(C=Data(numpy.ones((d,)),ReducedFunction(self.domain)))
1130 gross 1841 coeff=mypde.getCoefficient("C_reduced")
1131 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((d,),ReducedFunction(self.domain),1,1))
1132 gross 1072 def test_setCoefficient_D_reduced_Scalar_usingD(self):
1133     d=self.domain.getDim()
1134     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1135     mypde.setValue(D=Scalar(1.,ReducedFunction(self.domain)))
1136 gross 1841 coeff=mypde.getCoefficient("D_reduced")
1137 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((),ReducedFunction(self.domain),1,1))
1138 gross 1072 def test_setCoefficient_X_reduced_Scalar_usingX(self):
1139     d=self.domain.getDim()
1140     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1141 jfenwick 2455 mypde.setValue(X_reduced=Data(numpy.ones((d,)),ReducedFunction(self.domain)))
1142 gross 1841 coeff=mypde.getCoefficient("X_reduced")
1143 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((d,),ReducedFunction(self.domain),1))
1144 gross 1072 def test_setCoefficient_Y_reduced_Scalar_usingY(self):
1145     d=self.domain.getDim()
1146     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1147     mypde.setValue(Y=Scalar(1.,ReducedFunction(self.domain)))
1148 gross 1841 coeff=mypde.getCoefficient("Y_reduced")
1149 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((),ReducedFunction(self.domain),1))
1150 gross 1072 def test_setCoefficient_y_reduced_Scalar_using_y(self):
1151     d=self.domain.getDim()
1152     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1153     mypde.setValue(y=Scalar(1.,ReducedFunctionOnBoundary(self.domain)))
1154 gross 1841 coeff=mypde.getCoefficient("y_reduced")
1155 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((),ReducedFunctionOnBoundary(self.domain),1))
1156 gross 1072 def test_setCoefficient_d_reduced_Scalar_using_d(self):
1157     d=self.domain.getDim()
1158     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1159     mypde.setValue(d=Scalar(1.,ReducedFunctionOnBoundary(self.domain)))
1160 gross 1841 coeff=mypde.getCoefficient("d_reduced")
1161 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((),ReducedFunctionOnBoundary(self.domain),1,1))
1162 gross 1072 def test_setCoefficient_d_contact_reduced_Scalar_using_d_contact(self):
1163     d=self.domain.getDim()
1164     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1165     mypde.setValue(d_contact=Scalar(1.,ReducedFunctionOnContactZero(self.domain)))
1166 gross 1841 coeff=mypde.getCoefficient("d_contact_reduced")
1167 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((),ReducedFunctionOnContactZero(self.domain),1,1))
1168 gross 1072 def test_setCoefficient_y_contact_reduced_Scalar_using_y_contact(self):
1169     d=self.domain.getDim()
1170     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1171     mypde.setValue(y_contact=Scalar(1.,ReducedFunctionOnContactZero(self.domain)))
1172 gross 1841 coeff=mypde.getCoefficient("y_contact_reduced")
1173 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((),ReducedFunctionOnContactZero(self.domain),1))
1174 jgs 149 #
1175     # set coefficients for systems:
1176     #
1177     def test_setCoefficient_A_System(self):
1178     d=self.domain.getDim()
1179     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1180 jfenwick 2455 mypde.setValue(A=numpy.ones((self.N,d,self.N,d)))
1181 gross 1841 coeff=mypde.getCoefficient("A")
1182 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,d,self.N,d),Function(self.domain),self.N,self.N))
1183 jgs 149 def test_setCoefficient_B_System(self):
1184     d=self.domain.getDim()
1185     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1186 jfenwick 2455 mypde.setValue(B=numpy.ones((self.N,d,self.N)))
1187 gross 1841 coeff=mypde.getCoefficient("B")
1188 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,d,self.N),Function(self.domain),self.N,self.N))
1189 jgs 149 def test_setCoefficient_C_System(self):
1190     d=self.domain.getDim()
1191     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1192 jfenwick 2455 mypde.setValue(C=numpy.ones((self.N,self.N,d)))
1193 gross 1841 coeff=mypde.getCoefficient("C")
1194 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,self.N,d),Function(self.domain),self.N,self.N))
1195 jgs 149 def test_setCoefficient_D_System(self):
1196     d=self.domain.getDim()
1197     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1198 jfenwick 2455 mypde.setValue(D=numpy.ones((self.N,self.N)))
1199 gross 1841 coeff=mypde.getCoefficient("D")
1200 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,self.N),Function(self.domain),self.N,self.N))
1201 jgs 149 def test_setCoefficient_X_System(self):
1202     d=self.domain.getDim()
1203     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1204 jfenwick 2455 mypde.setValue(X=numpy.ones((self.N,d)))
1205 gross 1841 coeff=mypde.getCoefficient("X")
1206 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((self.N,d),Function(self.domain),self.N))
1207 jgs 149 def test_setCoefficient_Y_System(self):
1208     d=self.domain.getDim()
1209     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1210 jfenwick 2455 mypde.setValue(Y=numpy.ones((self.N,)))
1211 gross 1841 coeff=mypde.getCoefficient("Y")
1212 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((self.N,),Function(self.domain),self.N))
1213 jgs 149 def test_setCoefficient_y_System(self):
1214     d=self.domain.getDim()
1215     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1216 jfenwick 2455 mypde.setValue(y=numpy.ones((self.N,)))
1217 gross 1841 coeff=mypde.getCoefficient("y")
1218 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((self.N,),FunctionOnBoundary(self.domain),self.N))
1219 jgs 149 def test_setCoefficient_d_System(self):
1220     d=self.domain.getDim()
1221     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1222 jfenwick 2455 mypde.setValue(d=numpy.ones((self.N,self.N)))
1223 gross 1841 coeff=mypde.getCoefficient("d")
1224 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,self.N),FunctionOnBoundary(self.domain),self.N,self.N))
1225 jgs 149 def test_setCoefficient_d_contact_System(self):
1226     d=self.domain.getDim()
1227     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1228 jfenwick 2455 mypde.setValue(d_contact=numpy.ones((self.N,self.N)))
1229 gross 1841 coeff=mypde.getCoefficient("d_contact")
1230 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,self.N),FunctionOnContactZero(self.domain),self.N,self.N))
1231 jgs 149 def test_setCoefficient_y_contact_System(self):
1232     d=self.domain.getDim()
1233     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1234 jfenwick 2455 mypde.setValue(y_contact=numpy.ones((self.N,)))
1235 gross 1841 coeff=mypde.getCoefficient("y_contact")
1236 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((self.N,),FunctionOnContactZero(self.domain),self.N))
1237 gross 1072 def test_setCoefficient_A_reduced_System(self):
1238     d=self.domain.getDim()
1239     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1240 jfenwick 2455 mypde.setValue(A_reduced=numpy.ones((self.N,d,self.N,d)))
1241 gross 1841 coeff=mypde.getCoefficient("A_reduced")
1242 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,d,self.N,d),ReducedFunction(self.domain),self.N,self.N))
1243 gross 1072 def test_setCoefficient_B_reduced_System(self):
1244     d=self.domain.getDim()
1245     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1246 jfenwick 2455 mypde.setValue(B_reduced=numpy.ones((self.N,d,self.N)))
1247 gross 1841 coeff=mypde.getCoefficient("B_reduced")
1248 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,d,self.N),ReducedFunction(self.domain),self.N,self.N))
1249 gross 1072 def test_setCoefficient_C_reduced_System(self):
1250     d=self.domain.getDim()
1251     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1252 jfenwick 2455 mypde.setValue(C_reduced=numpy.ones((self.N,self.N,d)))
1253 gross 1841 coeff=mypde.getCoefficient("C_reduced")
1254 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,self.N,d),ReducedFunction(self.domain),self.N,self.N))
1255 gross 1072 def test_setCoefficient_D_System_reduced(self):
1256     d=self.domain.getDim()
1257     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1258 jfenwick 2455 mypde.setValue(D_reduced=numpy.ones((self.N,self.N)))
1259 gross 1841 coeff=mypde.getCoefficient("D_reduced")
1260 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,self.N),ReducedFunction(self.domain),self.N,self.N))
1261 gross 1072 def test_setCoefficient_X_System_reduced(self):
1262     d=self.domain.getDim()
1263     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1264 jfenwick 2455 mypde.setValue(X_reduced=numpy.ones((self.N,d)))
1265 gross 1841 coeff=mypde.getCoefficient("X_reduced")
1266 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((self.N,d),ReducedFunction(self.domain),self.N))
1267 gross 1072 def test_setCoefficient_Y_System_reduced(self):
1268     d=self.domain.getDim()
1269     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1270 jfenwick 2455 mypde.setValue(Y_reduced=numpy.ones((self.N,)))
1271 gross 1841 coeff=mypde.getCoefficient("Y_reduced")
1272 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((self.N,),ReducedFunction(self.domain),self.N))
1273 gross 1072 def test_setCoefficient_y_System_reduced(self):
1274     d=self.domain.getDim()
1275     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1276 jfenwick 2455 mypde.setValue(y_reduced=numpy.ones((self.N,)))
1277 gross 1841 coeff=mypde.getCoefficient("y_reduced")
1278 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((self.N,),ReducedFunctionOnBoundary(self.domain),self.N))
1279 gross 1072 def test_setCoefficient_d_reduced_System(self):
1280     d=self.domain.getDim()
1281     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1282 jfenwick 2455 mypde.setValue(d_reduced=numpy.ones((self.N,self.N)))
1283 gross 1841 coeff=mypde.getCoefficient("d_reduced")
1284 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,self.N),ReducedFunctionOnBoundary(self.domain),self.N,self.N))
1285 gross 1072 def test_setCoefficient_d_contact_reduced_System(self):
1286     d=self.domain.getDim()
1287     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1288 jfenwick 2455 mypde.setValue(d_contact_reduced=numpy.ones((self.N,self.N)))
1289 gross 1841 coeff=mypde.getCoefficient("d_contact_reduced")
1290 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,self.N),ReducedFunctionOnContactZero(self.domain),self.N,self.N))
1291 gross 1072 def test_setCoefficient_y_contact_reduced_System(self):
1292     d=self.domain.getDim()
1293     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1294 jfenwick 2455 mypde.setValue(y_contact_reduced=numpy.ones((self.N,)))
1295 gross 1841 coeff=mypde.getCoefficient("y_contact_reduced")
1296 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((self.N,),ReducedFunctionOnContactZero(self.domain),self.N))
1297 jgs 149 def test_setCoefficient_r_System(self):
1298     d=self.domain.getDim()
1299     mypde=LinearPDE(self.domain,numEquations=3,debug=self.DEBUG)
1300 jfenwick 2455 mypde.setValue(r=numpy.ones((self.N,)))
1301 gross 1841 coeff=mypde.getCoefficient("r")
1302 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions()),((self.N,),Solution(self.domain),self.N))
1303 jgs 149 def test_setCoefficient_q_System(self):
1304     d=self.domain.getDim()
1305     mypde=LinearPDE(self.domain,numEquations=3,debug=self.DEBUG)
1306 jfenwick 2455 mypde.setValue(q=numpy.ones((self.N,)))
1307 gross 1841 coeff=mypde.getCoefficient("q")
1308 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions()),((self.N,),Solution(self.domain),self.N))
1309 jgs 149 def test_setCoefficient_r_System_reducedOn(self):
1310     d=self.domain.getDim()
1311     mypde=LinearPDE(self.domain,numEquations=3,debug=self.DEBUG)
1312     mypde.setReducedOrderOn()
1313 jfenwick 2455 mypde.setValue(r=numpy.ones((self.N,)))
1314 gross 1841 coeff=mypde.getCoefficient("r")
1315 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions()),((self.N,),ReducedSolution(self.domain),self.N))
1316 jgs 149 def test_setCoefficient_q_System_reducedOn(self):
1317     d=self.domain.getDim()
1318     mypde=LinearPDE(self.domain,numEquations=3,debug=self.DEBUG)
1319     mypde.setReducedOrderOn()
1320 jfenwick 2455 mypde.setValue(q=numpy.ones((self.N,)))
1321 gross 1841 coeff=mypde.getCoefficient("q")
1322 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions()),((self.N,),ReducedSolution(self.domain),self.N))
1323 jgs 149
1324 gross 1072 def test_setCoefficient_A_reduced_System_using_A(self):
1325     d=self.domain.getDim()
1326     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1327 jfenwick 2455 mypde.setValue(A=Data(numpy.ones((self.N,d,self.N,d)),ReducedFunction(self.domain)))
1328 gross 1841 coeff=mypde.getCoefficient("A_reduced")
1329 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,d,self.N,d),ReducedFunction(self.domain),self.N,self.N))
1330 gross 1072 def test_setCoefficient_B_reduced_System_using_B(self):
1331     d=self.domain.getDim()
1332     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1333 jfenwick 2455 mypde.setValue(B=Data(numpy.ones((self.N,d,self.N)),ReducedFunction(self.domain)))
1334 gross 1841 coeff=mypde.getCoefficient("B_reduced")
1335 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,d,self.N),ReducedFunction(self.domain),self.N,self.N))
1336 gross 1072 def test_setCoefficient_C_reduced_System_using_C(self):
1337     d=self.domain.getDim()
1338     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1339 jfenwick 2455 mypde.setValue(C=Data(numpy.ones((self.N,self.N,d)),ReducedFunction(self.domain)))
1340 gross 1841 coeff=mypde.getCoefficient("C_reduced")
1341 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,self.N,d),ReducedFunction(self.domain),self.N,self.N))
1342 gross 1072 def test_setCoefficient_D_System_reduced_using_D(self):
1343     d=self.domain.getDim()
1344     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1345 jfenwick 2455 mypde.setValue(D=Data(numpy.ones((self.N,self.N)),ReducedFunction(self.domain)))
1346 gross 1841 coeff=mypde.getCoefficient("D_reduced")
1347 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,self.N),ReducedFunction(self.domain),self.N,self.N))
1348 gross 1072 def test_setCoefficient_X_System_reduced_using_X(self):
1349     d=self.domain.getDim()
1350     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1351 jfenwick 2455 mypde.setValue(X=Data(numpy.ones((self.N,d)),ReducedFunction(self.domain)))
1352 gross 1841 coeff=mypde.getCoefficient("X_reduced")
1353 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((self.N,d),ReducedFunction(self.domain),self.N))
1354 gross 1072 def test_setCoefficient_Y_System_reduced_using_Y(self):
1355     d=self.domain.getDim()
1356     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1357 jfenwick 2455 mypde.setValue(Y=Data(numpy.ones((self.N,)),ReducedFunction(self.domain)))
1358 gross 1841 coeff=mypde.getCoefficient("Y_reduced")
1359 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((self.N,),ReducedFunction(self.domain),self.N))
1360 gross 1072 def test_setCoefficient_y_reduced_System_using_y(self):
1361     d=self.domain.getDim()
1362     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1363 jfenwick 2455 mypde.setValue(y=Data(numpy.ones((self.N,)),ReducedFunctionOnBoundary(self.domain)))
1364 gross 1841 coeff=mypde.getCoefficient("y_reduced")
1365 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((self.N,),ReducedFunctionOnBoundary(self.domain),self.N))
1366 gross 1072 def test_setCoefficient_d_reduced_System_using_d(self):
1367     d=self.domain.getDim()
1368     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1369 jfenwick 2455 mypde.setValue(d=Data(numpy.ones((self.N,self.N)),ReducedFunctionOnBoundary(self.domain)))
1370 gross 1841 coeff=mypde.getCoefficient("d_reduced")
1371 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,self.N),ReducedFunctionOnBoundary(self.domain),self.N,self.N))
1372 gross 1072 def test_setCoefficient_d_contact_reduced_System_using_d_contact(self):
1373     d=self.domain.getDim()
1374     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1375 jfenwick 2455 mypde.setValue(d_contact=Data(numpy.ones((self.N,self.N)),ReducedFunctionOnContactZero(self.domain)))
1376 gross 1841 coeff=mypde.getCoefficient("d_contact_reduced")
1377 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumSolutions(), mypde.getNumEquations()),((self.N,self.N),ReducedFunctionOnContactZero(self.domain),self.N,self.N))
1378 gross 1072 def test_setCoefficient_y_contact_reduced_System_using_y_contact(self):
1379     d=self.domain.getDim()
1380     mypde=LinearPDE(self.domain,numSolutions=3,debug=self.DEBUG)
1381 jfenwick 2455 mypde.setValue(y_contact=Data(numpy.ones((self.N,)),ReducedFunctionOnContactZero(self.domain)))
1382 gross 1841 coeff=mypde.getCoefficient("y_contact_reduced")
1383 gross 2474 self.failUnlessEqual((coeff.getShape(),coeff.getFunctionSpace(), mypde.getNumEquations()),((self.N,),ReducedFunctionOnContactZero(self.domain),self.N))
1384 jgs 149 def test_resetCoefficient_HomogeneousConstraint(self):
1385     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1386     x=self.domain.getX()
1387 gross 304 mypde.setValue(A=kronecker(self.domain),Y=1.,q=whereZero(x[0]))
1388 jgs 149 u1=mypde.getSolution()
1389     mypde.setValue(Y=2.)
1390     u2=mypde.getSolution()
1391     self.failUnless(self.check(u2,2*u1),'solution is wrong.')
1392    
1393     def test_resetCoefficient_InHomogeneousConstraint(self):
1394     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1395 jgs 154 mypde.setSymmetryOn()
1396 gross 2474 mypde.getSolverOptions().setVerbosity(self.VERBOSE)
1397 jgs 149 x=self.domain.getX()
1398 gross 304 mypde.setValue(A=kronecker(self.domain),D=1.,Y=1.,r=1,q=whereZero(x[0]))
1399 gross 2474 u1=mypde.getSolution()
1400 jgs 149 mypde.setValue(Y=2.,D=2)
1401 gross 2474 u2=mypde.getSolution()
1402 jgs 154 self.failUnless(self.check(u2,u1),'first solution is wrong.')
1403 gross 2474 u2=mypde.getSolution()
1404 jgs 154 self.failUnless(self.check(u2,u1),'first solution is wrong.')
1405 jgs 149 mypde.setValue(r=2,Y=4.)
1406 gross 2474 u2=mypde.getSolution()
1407 jgs 154 self.failUnless(self.check(u2,2*u1),'second solution is wrong.')
1408 jgs 149
1409     def test_symmetryCheckTrue_System(self):
1410     d=self.domain.getDim()
1411     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1412 jfenwick 2455 A=numpy.ones((self.N,d,self.N,d))
1413     C=2*numpy.ones((self.N,self.N,d))
1414     B=2*numpy.ones((self.N,d,self.N))
1415     D=3*numpy.ones((self.N,self.N))
1416     d=4*numpy.ones((self.N,self.N))
1417     d_contact=5*numpy.ones((self.N,self.N))
1418 gross 1072 mypde.setValue(A=A,B=B,C=C,D=D,d=d,d_contact=d_contact,A_reduced=-A,B_reduced=-B,C_reduced=-C,D_reduced=-D,d_reduced=-d,d_contact_reduced=-d_contact)
1419 jgs 149 self.failUnless(mypde.checkSymmetry(verbose=False),"symmetry detected")
1420    
1421     def test_symmetryCheckFalse_A_System(self):
1422     d=self.domain.getDim()
1423     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1424 jfenwick 2455 A=numpy.ones((self.N,d,self.N,d))
1425 jgs 149 A[1,1,1,0]=0.
1426     mypde.setValue(A=A)
1427     self.failUnless(not mypde.checkSymmetry(verbose=False),"symmetry detected")
1428     def test_symmetryCheckFalse_BC_System(self):
1429     d=self.domain.getDim()
1430     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1431 jfenwick 2455 C=2*numpy.ones((self.N,self.N,d))
1432     B=2*numpy.ones((self.N,d,self.N))
1433 jgs 149 B[0,0,1]=1.
1434     mypde.setValue(B=B,C=C)
1435     self.failUnless(not mypde.checkSymmetry(verbose=False),"symmetry detected")
1436    
1437     def test_symmetryCheckFalse_D_System(self):
1438     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1439 jfenwick 2455 D=3*numpy.ones((self.N,self.N))
1440 jgs 149 D[0,1]=0.
1441     mypde.setValue(D=D)
1442     self.failUnless(not mypde.checkSymmetry(verbose=False),"symmetry detected")
1443    
1444     def test_symmetryCheckFalse_d_System(self):
1445     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1446 jfenwick 2455 d=4*numpy.ones((self.N,self.N))
1447 jgs 149 d[0,1]=0.
1448     mypde.setValue(d=d)
1449     self.failUnless(not mypde.checkSymmetry(verbose=False),"symmetry detected")
1450    
1451     def test_symmetryCheckFalse_d_contact_System(self):
1452     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1453 jfenwick 2455 d_contact=5*numpy.ones((self.N,self.N))
1454 jgs 149 d_contact[0,1]=0.
1455     mypde.setValue(d_contact=d_contact)
1456     self.failUnless(not mypde.checkSymmetry(verbose=False),"symmetry detected")
1457    
1458 gross 1072 def test_symmetryCheckFalse_A_reduced_System(self):
1459     d=self.domain.getDim()
1460     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1461 jfenwick 2455 A=numpy.ones((self.N,d,self.N,d))
1462 gross 1072 A[1,1,1,0]=0.
1463     mypde.setValue(A_reduced=A)
1464     self.failUnless(not mypde.checkSymmetry(verbose=False),"symmetry detected")
1465     def test_symmetryCheckFalse_BC_reduced_System(self):
1466     d=self.domain.getDim()
1467     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1468 jfenwick 2455 C=2*numpy.ones((self.N,self.N,d))
1469     B=2*numpy.ones((self.N,d,self.N))
1470 gross 1072 B[0,0,1]=1.
1471     mypde.setValue(B_reduced=B,C_reduced=C)
1472     self.failUnless(not mypde.checkSymmetry(verbose=False),"symmetry detected")
1473    
1474     def test_symmetryCheckFalse_D_reduced_System(self):
1475     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1476 jfenwick 2455 D=3*numpy.ones((self.N,self.N))
1477 gross 1072 D[0,1]=0.
1478     mypde.setValue(D_reduced=D)
1479     self.failUnless(not mypde.checkSymmetry(verbose=False),"symmetry detected")
1480    
1481     def test_symmetryCheckFalse_d_reduced_System(self):
1482     mypde=LinearPDE(self.domain,debug=self.DEBUG)
1483 jfenwick 2455 d=4*numpy.ones((self.N,self.N))
1484 gross 1072 d[0,1]=0.
1485     mypde.setValue