test/python/pdetoolsTest.py

# $Id$

from esys.escript import *
from esys.escript.pdetools import Locator,Projector,TimeIntegrationManager
from esys.finley import Rectangle

def testTimeIntegrationManager():
   t=0.
   dt=0.1
   tm=TimeIntegrationManager(0.,p=1)
   while t<1.:
      t+=dt
      tm.checkin(dt,t)
   v_guess=tm.extrapolate(dt)
   e=abs(v_guess-(tm.getTime()+dt))
   error_max,error_text=e,"testTimeIntegrationManager: scalar"

   t=0.
   dt=0.3
   tm=TimeIntegrationManager(0.,0.,p=1)
   while t<1.:
      t+=dt
      tm.checkin(dt,t,3*t)
   v_guess=tm.extrapolate(dt)
   e=max(abs(v_guess[0]-(tm.getTime()+dt)),abs(v_guess[1]-(tm.getTime()+dt)*3.))
   if e>error_max: error_max,error_text=e,"testTimeIntegrationManager: vector"

   return error_max,error_text
def testLocator(domain):
      """runs a few test of the Locator"""

      error_max=0.
      error_text=""
      x=domain.getX()
      l=Locator(domain,[1.,1.])
      print l.getFunctionSpace()
      print l.getId()

      l=Locator(ContinuousFunction(domain),[1.,1.])
      print l.getFunctionSpace()
      print l.getId()

      text="l.getX()"
      err=Lsup(l.getX()-numarray.ones((2,)))
      print text," error = ",err
      if err>=error_max: error_max,error_text=err,text

      text="l(x)"
      err=Lsup(l(x)-numarray.ones((2,)))
      print text," error = ",err
      if err>=error_max: error_max,error_text=err,text

      text="l(x[0]+x[1])"
      err=abs(l(x[0]+x[1])-2.)
      print text," error = ",err
      if err>=error_max: error_max,error_text=err,text

      return error_max,error_text
      
def testProjector(domain):
      """runs a few test of the Projector factory and returns the largest error plus a description of the test this error occured"""
      error_max=0.
      error_text=""
      x=ContinuousFunction(domain).getX()
      for f in [True,False]:
         p=Projector(domain,reduce=False,fast=f)
         for r in range(5):
            text="range %s , fast=%s"%(r,f)
            if r==0:
               td_ref=x[0]
            elif r==1:
               td_ref=x
            elif r==2:
               td_ref=[[11.,12.],[21,22.]]*(x[0]+x[1])
            elif r==3:
               td_ref=[[[111.,112.],[121,122.]],[[211.,212.],[221,222.]]]*(x[0]+x[1])
            elif r==3:
               td_ref=[[[[1111.,1112.],[1121,1122.]],[[1211.,1212.],[1221,1222.]]],[[[2111.,2112.],[2121,2122.]],[[2211.,2212.],[2221,2222.]]]]*(x[0]+x[1])
            td=p(td_ref.interpolate(Function(domain)))
            er=Lsup(td-td_ref)/Lsup(td_ref)
            print text," error = ",er
            if er>error_max:
                 error_max=er
                 error_text=text
      return error_max,error_text


txt=testLocator(Rectangle(56,61))
print "test Locator: ",txt

txt=testProjector(Rectangle(56,61))
print "test Projector: ",txt

print testTimeIntegrationManager()
1	# $Id$
2
3	from esys.escript import *
4	from esys.escript.pdetools import Locator,Projector,TimeIntegrationManager
5	from esys.finley import Rectangle
6
7	def testTimeIntegrationManager():
8	t=0.
9	dt=0.1
10	tm=TimeIntegrationManager(0.,p=1)
11	while t<1.:
12	t+=dt
13	tm.checkin(dt,t)
14	v_guess=tm.extrapolate(dt)
15	e=abs(v_guess-(tm.getTime()+dt))
16	error_max,error_text=e,"testTimeIntegrationManager: scalar"
17
18	t=0.
19	dt=0.3
20	tm=TimeIntegrationManager(0.,0.,p=1)
21	while t<1.:
22	t+=dt
23	tm.checkin(dt,t,3*t)
24	v_guess=tm.extrapolate(dt)
25	e=max(abs(v_guess[0]-(tm.getTime()+dt)),abs(v_guess[1]-(tm.getTime()+dt)*3.))
26	if e>error_max: error_max,error_text=e,"testTimeIntegrationManager: vector"
27
28	return error_max,error_text
29	def testLocator(domain):
30	"""runs a few test of the Locator"""
31
32	error_max=0.
33	error_text=""
34	x=domain.getX()
35	l=Locator(domain,[1.,1.])
36	print l.getFunctionSpace()
37	print l.getId()
38
39	l=Locator(ContinuousFunction(domain),[1.,1.])
40	print l.getFunctionSpace()
41	print l.getId()
42
43	text="l.getX()"
44	err=Lsup(l.getX()-numarray.ones((2,)))
45	print text," error = ",err
46	if err>=error_max: error_max,error_text=err,text
47
48	text="l(x)"
49	err=Lsup(l(x)-numarray.ones((2,)))
50	print text," error = ",err
51	if err>=error_max: error_max,error_text=err,text
52
53	text="l(x[0]+x[1])"
54	err=abs(l(x[0]+x[1])-2.)
55	print text," error = ",err
56	if err>=error_max: error_max,error_text=err,text
57
58	return error_max,error_text
59
60	def testProjector(domain):
61	"""runs a few test of the Projector factory and returns the largest error plus a description of the test this error occured"""
62	error_max=0.
63	error_text=""
64	x=ContinuousFunction(domain).getX()
65	for f in [True,False]:
66	p=Projector(domain,reduce=False,fast=f)
67	for r in range(5):
68	text="range %s , fast=%s"%(r,f)
69	if r==0:
70	td_ref=x[0]
71	elif r==1:
72	td_ref=x
73	elif r==2:
74	td_ref=[[11.,12.],[21,22.]]*(x[0]+x[1])
75	elif r==3:
76	td_ref=[[[111.,112.],[121,122.]],[[211.,212.],[221,222.]]]*(x[0]+x[1])
77	elif r==3:
78	td_ref=[[[[1111.,1112.],[1121,1122.]],[[1211.,1212.],[1221,1222.]]],[[[2111.,2112.],[2121,2122.]],[[2211.,2212.],[2221,2222.]]]]*(x[0]+x[1])
79	td=p(td_ref.interpolate(Function(domain)))
80	er=Lsup(td-td_ref)/Lsup(td_ref)
81	print text," error = ",er
82	if er>error_max:
83	error_max=er
84	error_text=text
85	return error_max,error_text
86
87
88	txt=testLocator(Rectangle(56,61))
89	print "test Locator: ",txt
90
91	txt=testProjector(Rectangle(56,61))
92	print "test Projector: ",txt
93
94	print testTimeIntegrationManager()
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision