Skip to content
Snippets Groups Projects
Commit 32140902 authored by Moritz Huebner's avatar Moritz Huebner Committed by Paul Lasky
Browse files

Resolve "Injection in frequency domain doesn't update time domain strain"

parent ad5b0cd6
No related branches found
No related tags found
No related merge requests found
......@@ -216,6 +216,7 @@ class InterferometerStrainData(object):
if not len(self.frequency_array) == len(frequency_domain_strain):
raise ValueError("The frequency_array and the set strain have different lengths")
self._frequency_domain_strain = frequency_domain_strain
self._time_domain_strain = None
def to_gwpy_timeseries(self):
"""
......
......@@ -366,6 +366,13 @@ class TestInterferometer(unittest.TestCase):
expected = self.injection_polarizations['plus'] + self.injection_polarizations['cross'] + original_strain
self.assertTrue(np.array_equal(expected, self.ifo.strain_data._frequency_domain_strain))
def test_inject_signal_from_waveform_polarizations_update_time_domain_strain(self):
original_td_strain = self.ifo.strain_data.time_domain_strain
self.ifo.get_detector_response = lambda x, params: x['plus'] + x['cross']
self.ifo.inject_signal_from_waveform_polarizations(parameters=self.parameters,
injection_polarizations=self.injection_polarizations)
self.assertFalse(np.array_equal(original_td_strain, self.ifo.strain_data.time_domain_strain))
def test_inject_signal_from_waveform_polarizations_meta_data(self):
self.ifo.get_detector_response = lambda x, params: x['plus'] + x['cross']
self.ifo.inject_signal_from_waveform_polarizations(parameters=self.parameters,
......
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