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_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_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
