From 1944b80bab090922ea9e8296c89b802cfb82de7c Mon Sep 17 00:00:00 2001
From: MoritzThomasHuebner <email@moritz-huebner.de>
Date: Tue, 4 Sep 2018 16:29:07 +1000
Subject: [PATCH] Defined repr test for the cubic spline
---
test/calibration_tests.py | 18 ++++++++++++++++--
1 file changed, 16 insertions(+), 2 deletions(-)
diff --git a/test/calibration_tests.py b/test/calibration_tests.py
index b481e2936..0e696fa11 100644
--- a/test/calibration_tests.py
+++ b/test/calibration_tests.py
@@ -20,14 +20,22 @@ class TestBaseClass(unittest.TestCase):
class TestCubicSpline(unittest.TestCase):
def setUp(self):
+ self.prefix = 'recalib_'
+ self.minimum_frequency = 20
+ self.maximum_frequency = 1024
+ self.n_points = 5
self.model = calibration.CubicSpline(
- prefix='recalib_', minimum_frequency=20, maximum_frequency=1024,
- n_points=5)
+ prefix=self.prefix, minimum_frequency=self.minimum_frequency,
+ maximum_frequency=self.maximum_frequency, n_points=self.n_points)
self.parameters = {'recalib_{}_{}'.format(param, ii): 0.0
for ii in range(5)
for param in ['amplitude', 'phase']}
def tearDown(self):
+ del self.prefix
+ del self.minimum_frequency
+ del self.maximum_frequency
+ del self.n_points
del self.model
del self.parameters
@@ -37,6 +45,12 @@ class TestCubicSpline(unittest.TestCase):
**self.parameters)
assert np.alltrue(cal_factor.real == np.ones_like(frequency_array))
+ def test_repr(self):
+ expected = 'CubicSpline(prefix={}, minimum_frequency={}, maximum_frequency={}, n_points={})'\
+ .format(self.prefix, self.minimum_frequency, self.maximum_frequency, self.n_points)
+ actual = repr(self.model)
+ self.assertEqual(expected, actual)
+
class TestCubicSplineRequiresFourNodes(unittest.TestCase):
--
GitLab