summaryrefslogtreecommitdiff
path: root/test/test_parabola.py
diff options
context:
space:
mode:
authorAnthony LaTorre <devnull@localhost>2013-11-08 22:03:49 -0600
committertlatorre <tlatorre@uchicago.edu>2021-05-09 08:42:39 -0700
commit8374553fcc23b127d0edd9d3f02239bc15d7d9d6 (patch)
tree63526ce7ca00eebb282d81f7c2a7831685efb717 /test/test_parabola.py
parent4d81f5272d94c6e3d7932d5e06146ec1fcc99450 (diff)
downloadchroma-8374553fcc23b127d0edd9d3f02239bc15d7d9d6.tar.gz
chroma-8374553fcc23b127d0edd9d3f02239bc15d7d9d6.tar.bz2
chroma-8374553fcc23b127d0edd9d3f02239bc15d7d9d6.zip
cleanup tests
Diffstat (limited to 'test/test_parabola.py')
-rw-r--r--test/test_parabola.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/test/test_parabola.py b/test/test_parabola.py
index 54708d7..fee8c0d 100644
--- a/test/test_parabola.py
+++ b/test/test_parabola.py
@@ -7,7 +7,7 @@ from numpy.testing import assert_almost_equal
class Test1D(unittest.TestCase):
def setUp(self):
self.x = numpy.array([[-1.0], [0.0], [1.0]])
- self.y = unumpy.uarray(([2.0, 1.0, 6.0], [0.1, 0.1, 0.1]))
+ self.y = unumpy.uarray([2.0, 1.0, 6.0], [0.1, 0.1, 0.1])
self.a = 1.0
self.b = numpy.array([2.0])
self.c = numpy.array([[3.0]])
@@ -25,9 +25,9 @@ class Test1D(unittest.TestCase):
assert_almost_equal(c[0,0].nominal_value, 3.0)
# Compare to ROOT TGraph fitting uncerts
- assert_almost_equal(a.std_dev(), 0.1)
- assert_almost_equal(b[0].std_dev(), 0.0707107)
- assert_almost_equal(c[0,0].std_dev(), 0.122474, decimal=5)
+ assert_almost_equal(a.std_dev, 0.1)
+ assert_almost_equal(b[0].std_dev, 0.0707107)
+ assert_almost_equal(c[0,0].std_dev, 0.122474, decimal=5)
class Test2D(unittest.TestCase):
@@ -46,7 +46,7 @@ class Test2D(unittest.TestCase):
+ x0 * self.b[0] + x1 * self.b[1] \
+ x0**2 * self.c[0,0] + x0 * x1 * self.c[0,1] \
+ x1 * x0 * self.c[1,0] + x1**2 * self.c[1,1]
- self.y[i] = ufloat((value, 0.1))
+ self.y[i] = ufloat(value, 0.1)
def test_parabola_eval(self):
y = parabola.parabola_eval(self.x, self.a, self.b, self.c)