sm/smbinterp
1
0
Fork 0
Code Issues Pull Requests Activity

if order doesn't matter, the test for baker stuff is back online

Browse Source
This commit is contained in:
Stephen McQuay 2011-01-26 22:52:39 -07:00
parent 53d555914e
commit ba848f6924
1 changed files with 26 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
test/baker.test.py
View File

@ -38,7 +38,6 @@ class TestSequenceFunctions(unittest.TestCase):
r = [[-1, -1], [0, 2], [1, -1]] r = [[-1, -1], [0, 2], [1, -1]]
result = baker.get_phis(X, r) result = baker.get_phis(X, r)
#result = [round(i, self.accuracy) for i in result]
right_answer = [1/3.0, 1/3.0, 1/3.0] right_answer = [1/3.0, 1/3.0, 1/3.0]
@ -78,15 +77,14 @@ class TestSequenceFunctions(unittest.TestCase):
S = grid.grid(self.all_points[size_of_simplex:size_of_simplex + extra_points], S = grid.grid(self.all_points[size_of_simplex:size_of_simplex + extra_points],
self.q[size_of_simplex:size_of_simplex + extra_points]) self.q[size_of_simplex:size_of_simplex + extra_points])
answer = baker.run_baker(self.X, R, S) answer = baker.run_baker(self.X, R, S)
print "\n", answer['abc']
a = answer['abc'][0] a = answer['abc'][0]
b = answer['abc'][1] b = answer['abc'][1]
c = answer['abc'][2] c = answer['abc'][2]
self.assertAlmostEqual(a, c) self.assertEqual(sorted((a,b,c)), sorted((0,0.0,1/3.)))
self.assertAlmostEqual(c, 0.0)
self.assertAlmostEqual(b, 1/3.0)
def testRunBaker_2(self): def testRunBaker_2(self):
size_of_simplex = 3 size_of_simplex = 3
@ -100,12 +98,11 @@ class TestSequenceFunctions(unittest.TestCase):
answer = baker.run_baker(self.X, R, S) answer = baker.run_baker(self.X, R, S)
a = answer['a'] a = answer['abc'][0]
b = answer['b'] b = answer['abc'][1]
c = answer['c'] c = answer['abc'][2]
self.assertAlmostEqual(a, c) self.assertEqual(sorted((a,b,c)), sorted((2/3.0, 2/3.0, 1/3.0)))
self.assertAlmostEqual(c, 2/3.0)
def testRunBaker_3(self): def testRunBaker_3(self):
size_of_simplex = 3 size_of_simplex = 3
@ -118,13 +115,17 @@ class TestSequenceFunctions(unittest.TestCase):
self.q[size_of_simplex:size_of_simplex + extra_points]) self.q[size_of_simplex:size_of_simplex + extra_points])
answer = baker.run_baker(self.X, R, S) answer = baker.run_baker(self.X, R, S)
a = answer['a']
b = answer['b']
c = answer['c']
self.assertAlmostEqual(a, 13/14.0) a = answer['abc'][0]
self.assertAlmostEqual(b, 2 / 7.0) b = answer['abc'][1]
self.assertAlmostEqual(c, 15/14.0) c = answer['abc'][2]
a,b,c = sorted((a,b,c))
aa, bb, cc = sorted((13/14., 2/7., 15/14.))
self.assertAlmostEqual(a,aa)
self.assertAlmostEqual(b,bb)
self.assertAlmostEqual(c,cc)
def testRunBaker_4(self): def testRunBaker_4(self):
size_of_simplex = 3 size_of_simplex = 3
@ -137,13 +138,16 @@ class TestSequenceFunctions(unittest.TestCase):
self.q[size_of_simplex:size_of_simplex + extra_points]) self.q[size_of_simplex:size_of_simplex + extra_points])
answer = baker.run_baker(self.X, R, S) answer = baker.run_baker(self.X, R, S)
a = answer['a'] a = answer['abc'][0]
b = answer['b'] b = answer['abc'][1]
c = answer['c'] c = answer['abc'][2]
self.assertAlmostEqual(a, 48/53.0) a,b,c = sorted((a,b,c))
self.assertAlmostEqual(b, 15/53.0) aa,bb,cc = sorted((48/53.0, 15/53.0, 54/53.0))
self.assertAlmostEqual(c, 54/53.0)
self.assertAlmostEqual(a, aa)
self.assertAlmostEqual(b, bb)
self.assertAlmostEqual(c, cc)
if __name__ == '__main__': if __name__ == '__main__':
suite = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions) suite = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions)