Skip to content
Snippets Groups Projects
Commit 02a2e75e authored by MoritzThomasHuebner's avatar MoritzThomasHuebner
Browse files

Added repr for Recalibrate base class

parent 41b0494d
No related branches found
No related tags found
1 merge request!170Resolve "Add more __repr__ methods"
......@@ -12,8 +12,9 @@ class TestBaseClass(unittest.TestCase):
del self.model
def test_repr(self):
expected = 'Recalibrate(prefix={})'.format('recalib_')
expected = 'Recalibrate(prefix={})'.format('\'recalib_\'')
actual = repr(self.model)
print(expected)
self.assertEqual(expected, actual)
def test_calibration_factor(self):
......
......@@ -22,7 +22,7 @@ class Recalibrate(object):
self.prefix = prefix
def __repr__(self):
return self.__class__.__name__ + '(prefix={})'.format(self.prefix)
return self.__class__.__name__ + '(prefix=\'{}\')'.format(self.prefix)
def get_calibration_factor(self, frequency_array, **params):
"""Apply calibration model
......
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