Пример #1
0
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))
Пример #2
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))
Пример #3
0
def test_fft():
    assert_eq(fft(darr), npfft.fft(nparr))
    assert_eq(fft(darr2, axis=0), npfft.fft(nparr, axis=0))
Пример #4
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))
Пример #5
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))
Пример #6
0
def test_fft():
    assert_eq(fft(darr), np.fft.fft(nparr))
    assert_eq(fft(darr2, axis=0), np.fft.fft(nparr, axis=0))
Пример #7
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)
Пример #8
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))
Пример #9
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))
Пример #10
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