diff --git a/tests/test_fft.py b/tests/test_fft.py deleted file mode 100644 index 928f155..0000000 --- a/tests/test_fft.py +++ /dev/null @@ -1,9 +0,0 @@ -import numpy as np -import xdas as xd -import xdas.fft as xfft - -class TestRFFT: - def test_with_non_dimensional(self): - da = xd.synthetics.wavelet_wavefronts() - da["latitude"] = ("distance", np.arange(da.sizes["distance"])) - xfft.rfft(da) diff --git a/tests/test_signal.py b/tests/test_signal.py index a61dc0d..92b40c5 100644 --- a/tests/test_signal.py +++ b/tests/test_signal.py @@ -5,8 +5,6 @@ import xdas import xdas.signal as xp from xdas.synthetics import wavelet_wavefronts -import tempfile -import os class TestSignal: diff --git a/xdas/fft.py b/xdas/fft.py index 2599ae4..3cd3312 100644 --- a/xdas/fft.py +++ b/xdas/fft.py @@ -55,7 +55,7 @@ def fft(da, n=None, dim={"last": "spectrum"}, norm=None, parallel=None): data = func(da.values) coords = { newdim if name == olddim else name: f if name == olddim else da.coords[name] - for name in da.coords if (da[name].dim != olddim or name == olddim) + for name in da.coords } dims = tuple(newdim if dim == olddim else dim for dim in da.dims) return DataArray(data, coords, dims, da.name, da.attrs) @@ -110,7 +110,7 @@ def rfft(da, n=None, dim={"last": "frequency"}, norm=None, parallel=None): data = func(da.values, n, axis, norm) coords = { newdim if name == olddim else name: f if name == olddim else da.coords[name] - for name in da.coords if (da[name].dim != olddim or name == olddim) + for name in da.coords } dims = tuple(newdim if dim == olddim else dim for dim in da.dims) return DataArray(data, coords, dims, da.name, da.attrs)