Manifest.hs

-- Do not edit! Automatically created with doctest-extract from src/Data/Array/Accelerate/FFTW/Manifest.hs
{-# LINE 22 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}

module Test.Data.Array.Accelerate.FFTW.Manifest where

import Data.Array.Accelerate.FFTW.Manifest
import Test.DocTest.Base
import qualified Test.DocTest.Driver as DocTest

{-# LINE 23 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}
import     qualified Data.Array.Accelerate.Utility.Lift.Exp as Exp
import     qualified Data.Array.Accelerate.Interpreter as AI
import     qualified Data.Array.Accelerate as A
import     Data.Array.Accelerate.Utility.Lift.Exp (expr)
import     Data.Array.Accelerate.Data.Complex (Complex((:+)))
import     Data.Array.Accelerate ((:.)((:.)))

roundComplex     :: A.Exp (Complex Float) -> A.Exp (Complex Int)
roundComplex     =
       Exp.modify (expr:+expr) $ \(r:+i) -> A.round r :+ (A.round i :: A.Exp Int)

roundComplexArray     ::
       (A.Shape sh) => A.Array sh (Complex Float) -> A.Array sh (Complex Int)
roundComplexArray     = AI.run1 (A.map roundComplex)

test :: DocTest.T ()
test = do
 DocTest.printPrefix "Data.Array.Accelerate.FFTW.Manifest:41: "
{-# LINE 41 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}
 DocTest.example(
{-# LINE 41 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}
    roundComplexArray $ dft (A.fromList (A.Z:.3:.3) [1000,0,0, 0,1000,0, 0,0,1000])
  )
  [ExpectedLine [LineChunk "Matrix (Z :. 3 :. 3)"],ExpectedLine [LineChunk "  [ 1000 :+ 0,        1000 :+ 0,        1000 :+ 0,"],ExpectedLine [LineChunk "    1000 :+ 0, (-500) :+ (-866),    (-500) :+ 866,"],ExpectedLine [LineChunk "    1000 :+ 0,    (-500) :+ 866, (-500) :+ (-866)]"]]
 DocTest.printPrefix "Data.Array.Accelerate.FFTW.Manifest:53: "
{-# LINE 53 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}
 DocTest.example(
{-# LINE 53 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}
    roundComplexArray $ idft $ dft (A.fromList (A.Z:.3:.3) [1000,0,0, 0,1000,0, 0,0,1000])
  )
  [ExpectedLine [LineChunk "Matrix (Z :. 3 :. 3)"],ExpectedLine [LineChunk "  [ 3000 :+ 0,    0 :+ 0,    0 :+ 0,"],ExpectedLine [LineChunk "       0 :+ 0, 3000 :+ 0,    0 :+ 0,"],ExpectedLine [LineChunk "       0 :+ 0,    0 :+ 0, 3000 :+ 0]"]]
 DocTest.printPrefix "Data.Array.Accelerate.FFTW.Manifest:66: "
{-# LINE 66 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}
 DocTest.example(
{-# LINE 66 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}
    roundComplexArray $ dft2d (A.fromList (A.Z:.3:.3) [1000,0,0, 0,0,0, 0,0,0])
  )
  [ExpectedLine [LineChunk "Matrix (Z :. 3 :. 3)"],ExpectedLine [LineChunk "  [ 1000 :+ 0, 1000 :+ 0, 1000 :+ 0,"],ExpectedLine [LineChunk "    1000 :+ 0, 1000 :+ 0, 1000 :+ 0,"],ExpectedLine [LineChunk "    1000 :+ 0, 1000 :+ 0, 1000 :+ 0]"]]
 DocTest.printPrefix "Data.Array.Accelerate.FFTW.Manifest:72: "
{-# LINE 72 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}
 DocTest.example(
{-# LINE 72 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}
    roundComplexArray $ dft2d (A.fromList (A.Z:.3:.3) [0,1000,0, 0,0,0, 0,0,0])
  )
  [ExpectedLine [LineChunk "Matrix (Z :. 3 :. 3)"],ExpectedLine [LineChunk "  [ 1000 :+ 0, (-500) :+ (-866), (-500) :+ 866,"],ExpectedLine [LineChunk "    1000 :+ 0, (-500) :+ (-866), (-500) :+ 866,"],ExpectedLine [LineChunk "    1000 :+ 0, (-500) :+ (-866), (-500) :+ 866]"]]
 DocTest.printPrefix "Data.Array.Accelerate.FFTW.Manifest:78: "
{-# LINE 78 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}
 DocTest.example(
{-# LINE 78 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}
    roundComplexArray $ dft2d (A.fromList (A.Z:.3:.3) [1000,0,0, 0,1000,0, 0,0,1000])
  )
  [ExpectedLine [LineChunk "Matrix (Z :. 3 :. 3)"],ExpectedLine [LineChunk "  [ 3000 :+ 0,    0 :+ 0,    0 :+ 0,"],ExpectedLine [LineChunk "       0 :+ 0,    0 :+ 0, 3000 :+ 0,"],ExpectedLine [LineChunk "       0 :+ 0, 3000 :+ 0,    0 :+ 0]"]]
 DocTest.printPrefix "Data.Array.Accelerate.FFTW.Manifest:90: "
{-# LINE 90 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}
 DocTest.example(
{-# LINE 90 "src/Data/Array/Accelerate/FFTW/Manifest.hs" #-}
    roundComplexArray $ idft2d $ dft2d (A.fromList (A.Z:.3:.3) [1000,0,0, 0,1000,0, 0,0,1000])
  )
  [ExpectedLine [LineChunk "Matrix (Z :. 3 :. 3)"],ExpectedLine [LineChunk "  [ 9000 :+ 0,    0 :+ 0,    0 :+ 0,"],ExpectedLine [LineChunk "       0 :+ 0, 9000 :+ 0,    0 :+ 0,"],ExpectedLine [LineChunk "       0 :+ 0,    0 :+ 0, 9000 :+ 0]"]]