Skip to content
Snippets Groups Projects
Commit 64f26c2b authored by MoritzThomasHuebner's avatar MoritzThomasHuebner
Browse files

Fixed an issue with the CubicSpline repr method

parent 02a2e75e
No related branches found
No related tags found
1 merge request!170Resolve "Add more __repr__ methods"
......@@ -14,7 +14,6 @@ class TestBaseClass(unittest.TestCase):
def test_repr(self):
expected = 'Recalibrate(prefix={})'.format('\'recalib_\'')
actual = repr(self.model)
print(expected)
self.assertEqual(expected, actual)
def test_calibration_factor(self):
......@@ -52,7 +51,7 @@ class TestCubicSpline(unittest.TestCase):
assert np.alltrue(cal_factor.real == np.ones_like(frequency_array))
def test_repr(self):
expected = 'CubicSpline(prefix={}, minimum_frequency={}, maximum_frequency={}, n_points={})'\
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)
......
......@@ -83,7 +83,7 @@ class CubicSpline(Recalibrate):
self.spline_points = np.logspace(np.log10(minimum_frequency), np.log10(maximum_frequency), n_points)
def __repr__(self):
return self.__class__.__name__ + '(prefix={}, minimum_frequency={}, maximum_frequency={}, n_points={})'\
return self.__class__.__name__ + '(prefix=\'{}\', minimum_frequency={}, maximum_frequency={}, n_points={})'\
.format(self.prefix, self.minimum_frequency, self.maximum_frequency, self.n_points)
def get_calibration_factor(self, frequency_array, **params):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment