# -*- coding: utf-8 -*-
##############################################################################
#
# Copyright (c) 2003-2018 by The University of Queensland
# http://www.uq.edu.au
#
# Primary Business: Queensland, Australia
# Licensed under the Apache License, version 2.0
# http://www.apache.org/licenses/LICENSE-2.0
#
# Development until 2012 by Earth Systems Science Computational Center (ESSCC)
# Development 2012-2013 by School of Earth Sciences
# Development from 2014 by Centre for Geoscience Computing (GeoComp)
#
##############################################################################
from __future__ import print_function, division
__copyright__="""Copyright (c) 2003-2018 by The University of Queensland
http://www.uq.edu.au
Primary Business: Queensland, Australia"""
__license__="""Licensed under the Apache License, version 2.0
http://www.apache.org/licenses/LICENSE-2.0"""
__url__="https://launchpad.net/escript-finley"
import esys.escriptcore.utestselect as unittest
from esys.escriptcore.testing import *
from esys.escript.modelframe import Model,Link,Simulation,ParameterSet,ESySXMLParser,DataSource
import math
from io import StringIO
from xml.dom import minidom
import numpy
import sys
class XMLDocumentTestCase(unittest.TestCase):
def setUp(self):
o1=ODETEST(debug=False)
o1.u=10
o2=ODETEST(debug=False)
o2.u=-10.
o1.f=Link(o2,"u")
o2.f=Link(o1,"u")
m=Messenger()
o1.dt=0.01
m.message=Link(o1)
s=Simulation([o1,o2,m],debug=False)
s.run()
output = StringIO()
s.writeXML(output)
output.seek(0)
outputList = output.readlines()
self.xmlList = outputList
def testFirstLine(self):
firstLine = self.xmlList[0]
self.assertEqual('\n', firstLine)
def testEsysHeader(self):
header = self.xmlList[1]
self.assertEqual('\n', header)
def testEsysFooter(self):
footer = self.xmlList[-1]
self.assertEqual('\n', footer)
def testSimulationHeader(self):
pass
def testSimulationFooter(self):
pass
class SimulationTestCase(unittest.TestCase):
def setUp(self):
o1=ODETEST(debug=False)
o1.u=10
o2=ODETEST(debug=False)
o2.u=-10.
o1.f=Link(o2,"u")
o2.f=Link(o1,"u")
m=Messenger()
o1.dt=0.01
m.message=Link(o1)
self.s=Simulation([o1,o2,m],debug=False)
self.s.run()
output = StringIO()
self.s.writeXML(output)
output.seek(0)
self.xml = output.read()
def testSimulation(self):
assert "99
class ModeltoDomTestCase(unittest.TestCase):
def _xml(self, modulename, modelname):
# returns a modelframe class, generated from the xml
return '''
a 0.9 f 128 u tend
1.0 u 10
tol
1e-08 dt
0.01 message current error = 9.516258e-01 a
0.9 f 127 u tend
1.0 u
-10.0 tol 1e-08 dt
0.1 message
current error = 1.904837e+01 message 127 message a
0.9 f 127 u tend
1.0 u
-10.0 tol 1e-08 dt
0.1 message
current error = 1.904837e+01 a
0.9 f 128 u tend
1.0 u 10
tol
1e-08 dt
0.01 message current error = 9.516258e-01 a
0.9 f 128 u tend
1.0 u 10
tol
1e-08 dt
0.01 message current error = 9.516258e-01
''' % (modulename, modelname)
def testModuleAttribute(self):
esysxml=ESySXMLParser(self._xml('run_xml', 'ODETEST'))
modeldoc=esysxml.parse()[0]
def testModuleAttributeFails(self):
try:
esysxml=ESySXMLParser(self._xml('a', 'b'))
modeldoc=esysxml.parse()[0]
except ImportError:
return # correct
assert False, "This test should have resulted in an ImportError"
class Messenger(Model):
def __init__(self, *args, **kwargs):
Model.__init__(self, *args, **kwargs)
self.declareParameter(message="none")
def doInitialization(self):
self.__t=0
#print "I start talking now!"
def doStepPostprocessing(self,dt):
self.__t+=dt
#print "Message (time %e) : %s "%(self.__t,self.message)
def doFinalization(self):
#print "I have no more to say!"
pass
class ODETEST(Model):
""" implements a solver for the ODE
du/dt=a*u+f(t)
we use a implicit euler scheme :
u_n-u_{n-1}= dt*a u_n + st*f(t_n)
to get u_n we run an iterative process
u_{n.k}=u_{n-1}+dt*(a u_{n.i-1} + f(t_n))
input for this model are step size dt, end time tend and a value for
a, f and initial value for u. we need also a tolerance tol for a
stopping criterion.
"""
def __init__(self, *args, **kwargs):
Model.__init__(self, *args, **kwargs)
self.declareParameter(tend=1.,dt=0.1,a=0.9,u=10.,f=0.,message="",tol=1.e-8)
def doInitialization(self):
self.__tn=0
self.__iter=0
def doStepPreprocessing(self,dt):
self.__iter=0
self.__u_last=self.u
def doStep(self,dt):
self.__iter+=1
self.__u_old=self.u
self.u=self.__u_last+dt*(self.a*self.__u_old+self.f)
def terminate(self):
if self.__iter<1:
return False
else:
return abs(self.__u_old-self.u)=self.tend
if __name__ == '__main__':
run_tests(__name__, exit_on_failure=True)