Skip to content

Commit

Permalink
Revert "Format."
Browse files Browse the repository at this point in the history
This reverts commit fa5c0c7.
  • Loading branch information
atrabattoni committed Nov 29, 2024
1 parent 3a3f6c7 commit fa3d209
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 2 deletions.
9 changes: 9 additions & 0 deletions tests/test_fft.py
Original file line number Diff line number Diff line change
@@ -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)
2 changes: 2 additions & 0 deletions tests/test_signal.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
import xdas
import xdas.signal as xp
from xdas.synthetics import wavelet_wavefronts
import tempfile
import os


class TestSignal:
Expand Down
4 changes: 2 additions & 2 deletions xdas/fft.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -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)

0 comments on commit fa3d209

Please sign in to comment.