def test_consistency_frequency_array_to_frequency_array(self):
new_sampling_frequency, new_duration = utils.get_sampling_frequency_and_duration_from_frequency_array(
self.frequency_array)
new_frequency_array = \
utils.create_frequency_series(sampling_frequency=new_sampling_frequency,
duration=new_duration)
self.assertTrue(np.allclose(self.frequency_array, new_frequency_array))
def setUp(self):
self.start_time = 1.3
self.sampling_frequency = 5
self.duration = 1.6
self.frequency_array = utils.create_frequency_series(sampling_frequency=self.sampling_frequency,
duration=self.duration)
self.time_array = utils.create_time_series(sampling_frequency=self.sampling_frequency,
duration=self.duration,
starting_time=self.start_time)
def test_frequency_array_setter(self):
new_sampling_frequency = 100
new_duration = 3
new_frequency_array = create_frequency_series(
sampling_frequency=new_sampling_frequency, duration=new_duration)
self.series.frequency_array = new_frequency_array
self.assertTrue(
np.array_equal(new_frequency_array, self.series.frequency_array))
self.assertLessEqual(
np.abs(new_sampling_frequency - self.series.sampling_frequency), 1)
self.assertAlmostEqual(new_duration, self.series.duration)
self.assertAlmostEqual(self.start_time, self.series.start_time)
def __init__(
self,
duration,
sampling_frequency,
frequency_domain_source_model=call_cupy_tf2,
waveform_arguments=None,
parameter_conversion=convert_to_lal_binary_neutron_star_parameters,
):
if waveform_arguments is None:
waveform_arguments = dict(minimum_frequency=10)
self.fdsm = frequency_domain_source_model
self.waveform_arguments = waveform_arguments
self.frequency_array = xp.asarray(
create_frequency_series(duration=duration,
sampling_frequency=sampling_frequency))
self.conversion = parameter_conversion
def test_frequency_array_from_init(self):
expected = create_frequency_series(
sampling_frequency=self.sampling_frequency, duration=self.duration)
self.assertTrue(np.array_equal(expected, self.series.frequency_array))
def test_create_frequency_array(self):
expected_frequency_array = np.array([0.0, 0.625, 1.25, 1.875, 2.5])
frequency_array = utils.create_frequency_series(
sampling_frequency=self.sampling_frequency, duration=self.duration)
self.assertTrue(np.allclose(expected_frequency_array, frequency_array))
