def test_find_closest_value_angle(self): fftx = np.asarray([1, 2, 3, 4, 5]) fft = FFT(fftx, fftx * 2, fftx * 3) self.assertEqual((1, 2, 3), fft.closest_value(0.))
def test_find_closest_value_noangle(self): fftx = np.asarray([1, 2, 3, 4, 5]) fft = FFT(fftx, fftx * 2) assert_equal((1, 2, None), fft.closest_value(0.))