• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• afnumpy-master
  • afnumpy
    • decorators.py
    • sparse
      • sputils.py
      • __init__.py
      • csr.py
      • base.py
    • fft.py
    • random.py
    • lib
      • arraypad.py
      • function_base.py
      • type_check.py
      • stride_tricks.py
      • shape_base.py
      • __init__.py
      • scimath.py
    • indexing.py
    • multiarray.py
    • __init__.py
    • core
      • function_base.py
      • fromnumeric.py
      • multiarray.py
      • shape_base.py
      • __init__.py
      • numeric.py
      • dtypes.py
    • linalg
      • linalg.py
      • __init__.py
    • private_utils.py
  • examples
    • phase_retrieval.py
  • LICENSE
  • setup.py
  • setup.cfg
  • .travis.yml
  • README.md
  • tests
    • test_linalg.py
    • test_jit.py
    • test_performance.py
    • test_ndarray.py
    • test_lib.py
    • asserts.py
    • test_fft.py
    • test_arrayfire.py
    • test_core.py
    • conftest.py
  • .gitlab-ci.yml

import afnumpy
import afnumpy.fft
import numpy
import numpy.fft
from asserts import *

def test_fft():    
    b = numpy.random.random((3,3))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fft(a), numpy.fft.fft(b))

    b = numpy.random.random((3,2))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fft(a), numpy.fft.fft(b))

    b = numpy.random.random((5,3,2))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fft(a), numpy.fft.fft(b))

    b = numpy.random.random((5,3,2))+numpy.random.random((5,3,2))*1.0j
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fft(a), numpy.fft.fft(b))

def test_ifft():    
    # Real to complex inverse fft not implemented in arrayfire
    # b = numpy.random.random((3,3))
    # a = afnumpy.array(b)
    # fassert(afnumpy.fft.ifft(a), numpy.fft.ifft(b))

    # b = numpy.random.random((3,2))
    # a = afnumpy.array(b)
    # fassert(afnumpy.fft.ifft(a), numpy.fft.ifft(b))

    # b = numpy.random.random((5,3,2))
    # a = afnumpy.array(b)
    # fassert(afnumpy.fft.ifft(a), numpy.fft.ifft(b))

    b = numpy.random.random((5,3,2))+numpy.random.random((5,3,2))*1.0j
#    b = numpy.ones((3,3))+numpy.zeros((3,3))*1.0j
    a = afnumpy.array(b)
    fassert(afnumpy.fft.ifft(a), numpy.fft.ifft(b))

def test_fft2():    
    b = numpy.random.random((3,3))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fft2(a), numpy.fft.fft2(b))

    b = numpy.random.random((3,2))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fft2(a), numpy.fft.fft2(b))

    b = numpy.random.random((5,3,2))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fft2(a), numpy.fft.fft2(b))

    b = numpy.random.random((5,3,2))+numpy.random.random((5,3,2))*1.0j
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fft2(a), numpy.fft.fft2(b))

def test_ifft2():    
    # Real to complex inverse fft not implemented in arrayfire
    # b = numpy.random.random((3,3))
    # a = afnumpy.array(b)
    # fassert(afnumpy.fft.ifft2(a), numpy.fft.ifft2(b))

    # b = numpy.random.random((3,2))
    # a = afnumpy.array(b)
    # fassert(afnumpy.fft.ifft2(a), numpy.fft.ifft2(b))

    # b = numpy.random.random((5,3,2))
    # a = afnumpy.array(b)
    # fassert(afnumpy.fft.ifft2(a), numpy.fft.ifft2(b))

    b = numpy.random.random((5,3,2))+numpy.random.random((5,3,2))*1.0j
#    b = numpy.ones((3,3))+numpy.zeros((3,3))*1.0j
    a = afnumpy.array(b)
    fassert(afnumpy.fft.ifft2(a), numpy.fft.ifft2(b))

def test_fftn():    
    b = numpy.random.random((3,3))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fftn(a), numpy.fft.fftn(b))

    b = numpy.random.random((3,2))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fftn(a), numpy.fft.fftn(b))

    b = numpy.random.random((5,3,2))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fftn(a), numpy.fft.fftn(b))

    b = numpy.random.random((5,3,2))+numpy.random.random((5,3,2))*1.0j
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fftn(a), numpy.fft.fftn(b))

    # Test shape argument
    b = numpy.random.random((3,3))
    a = afnumpy.array(b)
    s = (3,3)
    fassert(afnumpy.fft.fftn(a, s), numpy.fft.fftn(b, s))
    s = (3,6)
    fassert(afnumpy.fft.fftn(a, s), numpy.fft.fftn(b, s))
    s = (3,2)
    fassert(afnumpy.fft.fftn(a, s), numpy.fft.fftn(b, s))

def test_ifftn():    
    # Real to complex inverse fft not implemented in arrayfire
    # b = numpy.random.random((3,3))
    # a = afnumpy.array(b)
    # fassert(afnumpy.fft.ifftn(a), numpy.fft.ifftn(b))

    # b = numpy.random.random((3,2))
    # a = afnumpy.array(b)
    # fassert(afnumpy.fft.ifftn(a), numpy.fft.ifftn(b))

    # b = numpy.random.random((5,3,2))
    # a = afnumpy.array(b)
    # fassert(afnumpy.fft.ifftn(a), numpy.fft.ifftn(b))

    b = numpy.random.random((5,3,2))+numpy.random.random((5,3,2))*1.0j
#    b = numpy.ones((3,3))+numpy.zeros((3,3))*1.0j
    a = afnumpy.array(b)
    fassert(afnumpy.fft.ifftn(a), numpy.fft.ifftn(b))

    # Test shape argument
    b = numpy.random.random((3,3))
    a = afnumpy.array(b)
    s = (3,3)
    fassert(afnumpy.fft.ifftn(a, s), numpy.fft.ifftn(b, s))
    s = (3,6)
    fassert(afnumpy.fft.ifftn(a, s), numpy.fft.ifftn(b, s))
    s = (3,2)
    fassert(afnumpy.fft.ifftn(a, s), numpy.fft.ifftn(b, s))

def test_fftshift():
    b = numpy.random.random((3))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fftshift(a), numpy.fft.fftshift(b))
    b = numpy.random.random((3,3))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fftshift(a), numpy.fft.fftshift(b))
    b = numpy.random.random((3,3,3))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.fftshift(a), numpy.fft.fftshift(b))
    fassert(afnumpy.fft.fftshift(a,axes=0), numpy.fft.fftshift(b,axes=0))
    fassert(afnumpy.fft.fftshift(a,axes=1), numpy.fft.fftshift(b,axes=1))
    fassert(afnumpy.fft.fftshift(a,axes=2), numpy.fft.fftshift(b,axes=2))
    fassert(afnumpy.fft.fftshift(a,axes=(1,2)), numpy.fft.fftshift(b,axes=(1,2)))

def test_ifftshift():
    b = numpy.random.random((3))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.ifftshift(a), numpy.fft.ifftshift(b))
    b = numpy.random.random((3,3))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.ifftshift(a), numpy.fft.ifftshift(b))
    b = numpy.random.random((3,3,3))
    a = afnumpy.array(b)
    fassert(afnumpy.fft.ifftshift(a), numpy.fft.ifftshift(b))
    fassert(afnumpy.fft.ifftshift(a,axes=0), numpy.fft.ifftshift(b,axes=0))
    fassert(afnumpy.fft.ifftshift(a,axes=1), numpy.fft.ifftshift(b,axes=1))
    fassert(afnumpy.fft.ifftshift(a,axes=2), numpy.fft.ifftshift(b,axes=2))
    fassert(afnumpy.fft.ifftshift(a,axes=(1,2)), numpy.fft.ifftshift(b,axes=(1,2)))