def test_time_array_from_init(self):
expected = create_time_series(
sampling_frequency=self.sampling_frequency,
duration=self.duration,
starting_time=self.start_time,
)
self.assertTrue(np.array_equal(expected, self.series.time_array))
def test_create_time_array(self):
expected_time_array = np.array(
[1.3, 1.5, 1.7, 1.9, 2.1, 2.3, 2.5, 2.7])
time_array = utils.create_time_series(
sampling_frequency=self.sampling_frequency,
duration=self.duration,
starting_time=self.start_time)
self.assertTrue(np.allclose(expected_time_array, time_array))
def test_consistency_time_array_to_time_array(self):
new_sampling_frequency, new_duration = \
utils.get_sampling_frequency_and_duration_from_time_array(self.time_array)
new_start_time = self.time_array[0]
new_time_array = utils.create_time_series(sampling_frequency=new_sampling_frequency,
duration=new_duration,
starting_time=new_start_time)
self.assertTrue(np.allclose(self.time_array, new_time_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_nfft_sine_function(self):
injected_frequency = 2.7324
duration = 100
times = utils.create_time_series(self.sampling_frequency, duration)
time_domain_strain = np.sin(2 * np.pi * times * injected_frequency + 0.4)
frequency_domain_strain, frequencies = bilby.core.utils.nfft(time_domain_strain, self.sampling_frequency)
frequency_at_peak = frequencies[np.argmax(np.abs(frequency_domain_strain))]
self.assertAlmostEqual(injected_frequency, frequency_at_peak, places=1)
def test_time_array_setter(self):
new_sampling_frequency = 100
new_duration = 3
new_start_time = 4
new_time_array = create_time_series(
sampling_frequency=new_sampling_frequency,
duration=new_duration,
starting_time=new_start_time,
)
self.series.time_array = new_time_array
self.assertTrue(np.array_equal(new_time_array, self.series.time_array))
self.assertAlmostEqual(
new_sampling_frequency, self.series.sampling_frequency, places=1
)
self.assertAlmostEqual(new_duration, self.series.duration, places=1)
self.assertAlmostEqual(new_start_time, self.series.start_time, places=1)
def test_illegal_sampling_frequency_and_duration(self):
with self.assertRaises(
utils.IllegalDurationAndSamplingFrequencyException):
_ = utils.create_time_series(sampling_frequency=7.7,
duration=1.3,
starting_time=0)
