55 lines
1.4 KiB
Python
Executable File
55 lines
1.4 KiB
Python
Executable File
#!/usr/bin/python
|
|
|
|
import unittest
|
|
|
|
from baker import run_baker
|
|
from baker.tools import exact_func
|
|
|
|
from grid.DD import grid
|
|
from grid.simplex import contains
|
|
|
|
|
|
class TestSequenceFunctions(unittest.TestCase):
|
|
def setUp(self):
|
|
self.points = [
|
|
[ 0.25, 0.40], # 0
|
|
[ 0.60, 0.80], # 1
|
|
[ 0.65, 0.28], # 2
|
|
[ 0.28, 0.65], # 3
|
|
[ 1.00, 0.75], # 4
|
|
[ 0.30, 0.95], # 5
|
|
[ 0.80, 0.50], # 6
|
|
[ 0.35, 0.15], # 7
|
|
]
|
|
self.q = [exact_func(p) for p in self.points]
|
|
|
|
self.X = [0.55, 0.45]
|
|
self.X = [0.25, 0.4001]
|
|
|
|
self.g = grid(self.points, self.q)
|
|
self.g.construct_connectivity()
|
|
self.R = self.g.create_mesh(range(3))
|
|
self.S = self.g.create_mesh(range(3,len(self.points)))
|
|
|
|
self.exact = exact_func(self.X)
|
|
|
|
self.answer = run_baker(self.X, self.R, self.S)
|
|
|
|
self.accuracy = 8
|
|
|
|
def test_R_contains_X(self):
|
|
self.assertTrue(contains(self.X, self.R.points))
|
|
|
|
def test_RunBaker(self):
|
|
print
|
|
print "X\n", self.X
|
|
print "R\n", self.R
|
|
print "S\n", self.S
|
|
print "exact\n",self.exact
|
|
print "qlin\n",self.answer['qlin']
|
|
self.assertTrue(self.answer)
|
|
|
|
if __name__ == '__main__':
|
|
suite = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions)
|
|
unittest.TextTestRunner(verbosity=3).run(suite)
|