From 02a2e75ea001e5db6006ab2b39b0bd5c009a8c09 Mon Sep 17 00:00:00 2001
From: MoritzThomasHuebner <email@moritz-huebner.de>
Date: Tue, 4 Sep 2018 16:33:34 +1000
Subject: [PATCH] Added repr for Recalibrate base class
---
test/calibration_tests.py | 3 ++-
tupak/gw/calibration.py | 2 +-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/test/calibration_tests.py b/test/calibration_tests.py
index 1c89f79b0..a916c841a 100644
--- a/test/calibration_tests.py
+++ b/test/calibration_tests.py
@@ -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):
diff --git a/tupak/gw/calibration.py b/tupak/gw/calibration.py
index 2c701593f..7460eb424 100644
--- a/tupak/gw/calibration.py
+++ b/tupak/gw/calibration.py
@@ -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
--
GitLab