Coverage for gpaw/gpu/cpupy/fft.py: 100%

1import numpy as np 

2 

3 

4def fftshift(a_Q, axes): 

5 from gpaw.gpu.cpupy import ndarray 

6 return ndarray(np.fft.fftshift(a_Q._data, axes=axes)) 

7 

8 

9def ifftshift(a_Q, axes): 

10 from gpaw.gpu.cpupy import ndarray 

11 return ndarray(np.fft.ifftshift(a_Q._data, axes=axes))