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)