diff --git a/test/calibration_tests.py b/test/calibration_tests.py index b481e2936ed88ec6d4cf68136c495f0ef4d25539..0e696fa115276182b4b5c1d93244ddbde48963be 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):