def test_fft_n_kwarg(): assert_eq(fft(darr, 5), npfft.fft(nparr, 5)) assert_eq(fft(darr, 13), npfft.fft(nparr, 13)) assert_eq(fft(darr2, 5, axis=0), npfft.fft(nparr, 5, axis=0)) assert_eq(fft(darr2, 13, axis=0), npfft.fft(nparr, 13, axis=0))
def test_cant_fft_chunked_axis(): bad_darr = da.from_array(nparr, chunks=(5, 5)) pytest.raises(ValueError, lambda: fft(bad_darr)) pytest.raises(ValueError, lambda: fft(bad_darr, axis=0))
def test_fft(): assert_eq(fft(darr), npfft.fft(nparr)) assert_eq(fft(darr2, axis=0), npfft.fft(nparr, axis=0))
def test_fft_n_kwarg(): assert_eq(fft(darr, 5), np.fft.fft(nparr, 5)) assert_eq(fft(darr, 13), np.fft.fft(nparr, 13)) assert_eq(fft(darr2, 5, axis=0), np.fft.fft(nparr, 5, axis=0)) assert_eq(fft(darr2, 13, axis=0), np.fft.fft(nparr, 13, axis=0))
def test_fft_consistent_names(): assert same_keys(fft(darr, 5), fft(darr, 5)) assert same_keys(fft(darr2, 5, axis=0), fft(darr2, 5, axis=0)) assert not same_keys(fft(darr, 5), fft(darr, 13))
def test_fft(): assert_eq(fft(darr), np.fft.fft(nparr)) assert_eq(fft(darr2, axis=0), np.fft.fft(nparr, axis=0))
def test_cant_fft_chunked_axis(): bad_darr = da.from_array(nparr, chunks=(5, 5)) with pytest.raises(ValueError): fft(bad_darr) with pytest.raises(ValueError): fft(bad_darr, axis=0)
def test_cant_fft_chunked_axis(): bad_darr = da.from_array(nparr, chunks=(5, 5)) assert raises(ValueError, lambda: fft(bad_darr)) assert raises(ValueError, lambda: fft(bad_darr, axis=0))
def filtered_fft(x, t_step, f_cutoff): x_freq = fft.fft(x) f = fft.fftfreq(x.shape[-1]) / t_step / 1000 # in kHz x_freq_filtered = x_freq[:, :, np.abs(f) < f_cutoff] return x_freq_filtered