From fa3d209fd1f0c82a1ba5fe423e998bbd6ab547bd Mon Sep 17 00:00:00 2001 From: Alister Trabattoni Date: Fri, 29 Nov 2024 16:01:45 +0100 Subject: [PATCH] Revert "Format." This reverts commit fa5c0c7fdf47d2a0a2ab5b0ce24b8129bd47e69b. --- tests/test_fft.py | 9 +++++++++ tests/test_signal.py | 2 ++ xdas/fft.py | 4 ++-- 3 files changed, 13 insertions(+), 2 deletions(-) create mode 100644 tests/test_fft.py diff --git a/tests/test_fft.py b/tests/test_fft.py new file mode 100644 index 0000000..928f155 --- /dev/null +++ b/tests/test_fft.py @@ -0,0 +1,9 @@ +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 92b40c5..a61dc0d 100644 --- a/tests/test_signal.py +++ b/tests/test_signal.py @@ -5,6 +5,8 @@ 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 3cd3312..2599ae4 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 + for name in da.coords if (da[name].dim != olddim or name == olddim) } 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 + for name in da.coords if (da[name].dim != olddim or name == olddim) } dims = tuple(newdim if dim == olddim else dim for dim in da.dims) return DataArray(data, coords, dims, da.name, da.attrs)