From 2fac76bafad3d7aa2c25bc77fa310f9c59590f7b Mon Sep 17 00:00:00 2001
From: MoritzThomasHuebner <email@moritz-huebner.de>
Date: Fri, 13 Jul 2018 06:13:56 +1000
Subject: [PATCH] Fixed the tests so they work properly with my changes
---
test/detector_tests.py | 14 ++++++++------
1 file changed, 8 insertions(+), 6 deletions(-)
diff --git a/test/detector_tests.py b/test/detector_tests.py
index 4b6abf966..b7d8f933c 100644
--- a/test/detector_tests.py
+++ b/test/detector_tests.py
@@ -30,7 +30,7 @@ class TestDetector(unittest.TestCase):
xarm_azimuth=self.xarm_azimuth, yarm_azimuth=self.yarm_azimuth,
xarm_tilt=self.xarm_tilt, yarm_tilt=self.yarm_tilt)
self.ifo.strain_data.set_from_frequency_domain_strain(
- [1], sampling_frequency=1, duration=1)
+ np.linspace(0, 4096, 4097), sampling_frequency=4096, duration=2)
def tearDown(self):
del self.name
@@ -214,7 +214,7 @@ class TestDetector(unittest.TestCase):
self.minimum_frequency = 10
self.maximum_frequency = 20
#self.ifo.frequency_array = np.array([8, 12, 16, 20, 24])
- plus = np.array([1, 2, 3, 4, 5])
+ plus = np.linspace(0, 4096, 4097)
response = self.ifo.get_detector_response(
waveform_polarizations=dict(plus=plus),
parameters=dict(ra=0, dec=0, geocent_time=0, psi=0))
@@ -227,11 +227,13 @@ class TestDetector(unittest.TestCase):
self.minimum_frequency = 10
self.maximum_frequency = 20
#self.ifo.frequency_array = np.array([8, 12, 16, 20, 24])
- plus = np.array([1, 2, 3, 4, 5])
+ plus = np.linspace(0, 4096, 4097)
response = self.ifo.get_detector_response(
waveform_polarizations=dict(plus=plus),
parameters=dict(ra=0, dec=0, geocent_time=0, psi=0))
- self.assertTrue(np.array_equal(response, plus*self.ifo.frequency_mask*np.exp(-1j*2*np.pi*self.ifo.frequency_array)))
+ expected_response = plus*self.ifo.frequency_mask*np.exp(-1j*2*np.pi*self.ifo.frequency_array)
+ self.assertTrue(np.allclose(abs(response),
+ abs(plus*self.ifo.frequency_mask*np.exp(-1j*2*np.pi*self.ifo.frequency_array))))
def test_get_detector_response_multiple_modes(self):
self.ifo.antenna_response = MagicMock(return_value=1)
@@ -240,8 +242,8 @@ class TestDetector(unittest.TestCase):
self.minimum_frequency = 10
self.maximum_frequency = 20
#self.ifo.frequency_array = np.array([8, 12, 16, 20, 24])
- plus = np.array([1, 2, 3, 4, 5])
- cross = np.array([6, 7, 8, 9, 10])
+ plus = np.linspace(0, 4096, 4097)
+ cross = np.linspace(0, 4096, 4097)
response = self.ifo.get_detector_response(
waveform_polarizations=dict(plus=plus, cross=cross),
parameters=dict(ra=0, dec=0, geocent_time=0, psi=0))
--
GitLab