From 562b6d7cea38e5da58b7bb0265c8488ff5d788e7 Mon Sep 17 00:00:00 2001
From: moritz <email@moritz-huebner.de>
Date: Tue, 1 May 2018 11:08:46 +1000
Subject: [PATCH] Moritz Huebner: Renamed source_model ->
 frequency_domain_source_model

---
 peyote/waveform_generator.py        | 4 ++--
 test/waveform_generator_tests.py    | 9 ++++++---
 tutorials/BasicTutorial.py          | 2 +-
 tutorials/custom_source_tutorial.py | 2 +-
 tutorials/make_standard_data.py     | 2 +-
 5 files changed, 11 insertions(+), 8 deletions(-)

diff --git a/peyote/waveform_generator.py b/peyote/waveform_generator.py
index 5694be289..ed79b6607 100644
--- a/peyote/waveform_generator.py
+++ b/peyote/waveform_generator.py
@@ -23,11 +23,11 @@ class WaveformGenerator(object):
 
     """
 
-    def __init__(self, source_model, sampling_frequency=4096, time_duration=1,
+    def __init__(self, frequency_domain_source_model, sampling_frequency=4096, time_duration=1,
                  parameters=None):
         self.time_duration = time_duration
         self.sampling_frequency = sampling_frequency
-        self.source_model = source_model
+        self.source_model = frequency_domain_source_model
         self.parameters = parameters
         self.__frequency_array_updated = False
         self.__time_array_updated = False
diff --git a/test/waveform_generator_tests.py b/test/waveform_generator_tests.py
index 26fed1a83..3ab64e404 100644
--- a/test/waveform_generator_tests.py
+++ b/test/waveform_generator_tests.py
@@ -17,7 +17,8 @@ def gaussian_frequency_domain_strain_2(frequency_array, a, m, s, ra, dec, geocen
 class TestWaveformGeneratorInstantiationWithoutOptionalParameters(unittest.TestCase):
 
     def setUp(self):
-        self.waveform_generator = peyote.waveform_generator.WaveformGenerator(source_model=gaussian_frequency_domain_strain)
+        self.waveform_generator = peyote.waveform_generator.WaveformGenerator(
+            frequency_domain_source_model=gaussian_frequency_domain_strain)
         self.simulation_parameters = dict(amplitude=1e-21, mu=100, sigma=1,
                                      ra=1.375,
                                      dec=-1.2108,
@@ -51,7 +52,8 @@ class TestWaveformGeneratorInstantiationWithoutOptionalParameters(unittest.TestC
 class TestParameterSetter(unittest.TestCase):
 
     def setUp(self):
-        self.waveform_generator = peyote.waveform_generator.WaveformGenerator(source_model=gaussian_frequency_domain_strain)
+        self.waveform_generator = peyote.waveform_generator.WaveformGenerator(
+            frequency_domain_source_model=gaussian_frequency_domain_strain)
         self.simulation_parameters = dict(amplitude=1e-21, mu=100, sigma=1,
                                      ra=1.375,
                                      dec=-1.2108,
@@ -89,7 +91,8 @@ class TestParameterSetter(unittest.TestCase):
 class TestSourceModelSetter(unittest.TestCase):
 
     def setUp(self):
-        self.waveform_generator = peyote.waveform_generator.WaveformGenerator(source_model=gaussian_frequency_domain_strain)
+        self.waveform_generator = peyote.waveform_generator.WaveformGenerator(
+            frequency_domain_source_model=gaussian_frequency_domain_strain)
         self.waveform_generator.source_model = gaussian_frequency_domain_strain_2
         self.simulation_parameters = dict(a=1e-21, m=100, s=1,
                                      ra=1.375,
diff --git a/tutorials/BasicTutorial.py b/tutorials/BasicTutorial.py
index 8bc4c195d..bc250d9e1 100644
--- a/tutorials/BasicTutorial.py
+++ b/tutorials/BasicTutorial.py
@@ -32,7 +32,7 @@ injection_parameters = dict(
 waveform_generator = peyote.waveform_generator.WaveformGenerator(
     sampling_frequency=sampling_frequency,
     time_duration=time_duration,
-    source_model=peyote.source.lal_binary_black_hole,
+    frequency_domain_source_model=peyote.source.lal_binary_black_hole,
     parameters=injection_parameters)
 hf_signal = waveform_generator.frequency_domain_strain()
 
diff --git a/tutorials/custom_source_tutorial.py b/tutorials/custom_source_tutorial.py
index 876537deb..982c550f4 100644
--- a/tutorials/custom_source_tutorial.py
+++ b/tutorials/custom_source_tutorial.py
@@ -73,7 +73,7 @@ injection_parameters = dict(amplitude=1e-21,
 sampling_parameters = peyote.prior.parse_floats_to_fixed_priors(injection_parameters)
 
 wg = peyote.waveform_generator.WaveformGenerator(
-     source_model=gaussian_frequency_domain_strain,
+     frequency_domain_source_model=gaussian_frequency_domain_strain,
      parameters=injection_parameters
      )
 
diff --git a/tutorials/make_standard_data.py b/tutorials/make_standard_data.py
index 8bdd58516..eeee0ebdc 100644
--- a/tutorials/make_standard_data.py
+++ b/tutorials/make_standard_data.py
@@ -32,7 +32,7 @@ simulation_parameters = dict(
     psi=2.659
 )
 #sampling_parameters = peyote.parameter.Parameter.parse_floats_to_parameters(simulation_parameters)
-waveform_generator = WaveformGenerator(source_model=peyote.source.lal_binary_black_hole,
+waveform_generator = WaveformGenerator(frequency_domain_source_model=peyote.source.lal_binary_black_hole,
                                        sampling_frequency=sampling_frequency,
                                        time_duration=time_duration,
                                        parameters=simulation_parameters)
-- 
GitLab