massiv-test-0.1.0: tests/Data/Massiv/Array/Ops/SliceSpec.hs
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE MonoLocalBinds #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE TypeFamilies #-}
module Data.Massiv.Array.Ops.SliceSpec (spec) where
import Control.Applicative ((<|>))
import Control.Exception
import Data.Massiv.Array.Unsafe
import Data.Massiv.Array as A
import Test.Massiv.Core
-----------
-- Size --
-----------
prop_ExtractEqualsExtractFromTo ::
( Eq (Array (R r) ix e)
, Show (Array (R r) ix e)
, Extract r ix e
)
=> proxy (r, ix, e)
-> SzIx ix
-> Array r ix e
-> Property
prop_ExtractEqualsExtractFromTo _ (SzIx (Sz eIx) sIx) arr =
(extractFromToM sIx eIx arr <|> Nothing) === extractM sIx (Sz (liftIndex2 (-) eIx sIx)) arr
specSizeN ::
( Eq (Array (R r) ix e)
, Show (Array (R r) ix e)
, Arbitrary (Array r ix e)
, Show (Array r ix e)
, Arbitrary ix
, Extract r ix e
)
=> proxy (r, ix, e)
-> Spec
specSizeN proxy =
describe "extract" $
it "ExtractEqualsExtractFromTo" $ property $ prop_ExtractEqualsExtractFromTo proxy
-----------
-- Slice --
-----------
prop_SliceRight ::
(Slice r ix e, OuterSlice r ix e, Eq (Elt r ix e), Show (Elt r ix e))
=> proxy (r, ix, e)
-> Int
-> Array r ix e
-> Property
prop_SliceRight _ i arr =
either (Left . displayException) Right (arr !?> i) ===
either (Left . displayException) Right (arr <!?> (dimensions (size arr), i))
prop_SliceLeft ::
(Slice r ix e, InnerSlice r ix e, Eq (Elt r ix e), Show (Elt r ix e))
=> proxy (r, ix, e)
-> Int
-> Array r ix e
-> Property
prop_SliceLeft _ i arr =
either (Left . displayException) Right (arr <!? i) ===
either (Left . displayException) Right (arr <!?> (1, i))
prop_SliceIndexDim2D :: ArrIx D Ix2 Int -> Property
prop_SliceIndexDim2D (ArrIx arr ix@(i :. j)) =
val === evaluate' (arr <! j) i .&&.
val === evaluate' (arr !> i) j
where
val = unsafeIndex arr ix
prop_SliceIndexDim2RankD :: ArrIx D Ix2 Int -> Property
prop_SliceIndexDim2RankD (ArrIx arr ix@(i :. j)) =
val === evaluate' (arr <!> (2, i)) j .&&.
val === evaluate' (arr <!> (1, j)) i
where
val = unsafeIndex arr ix
prop_SliceIndexDim3D :: ArrIx D Ix3 Int -> Property
prop_SliceIndexDim3D (ArrIx arr ix@(i :> j :. k)) =
val === evaluate' (arr <! k <! j) i .&&.
val === evaluate' (arr !> i !> j) k .&&.
val === evaluate' (arr <! k !> i) j .&&.
val === evaluate' (arr !> i <! k) j
where
val = unsafeIndex arr ix
prop_SliceIndexDim3RankD :: ArrIx D Ix3 Int -> Property
prop_SliceIndexDim3RankD (ArrIx arr ix@(i :> j :. k)) =
val === evaluate' (arr <!> (3, i) <!> (2, j)) k .&&.
val === evaluate' (arr <!> (3, i) <!> (1, k)) j .&&.
val === evaluate' (arr <!> (2, j) <!> (2, i)) k .&&.
val === evaluate' (arr <!> (2, j) <!> (1, k)) i .&&.
val === evaluate' (arr <!> (1, k) <!> (2, i)) j .&&.
val === evaluate' (arr <!> (1, k) <!> (1, j)) i
where
val = unsafeIndex arr ix
prop_SliceIndexDim2M :: ArrIx P Ix2 Int -> Property
prop_SliceIndexDim2M (ArrIx arr' ix@(i :. j)) =
val === (arr !> i ! j) .&&.
val === (arr <! j ! i)
where
arr = toManifest arr'
val = unsafeIndex arr ix
prop_SliceIndexDim2RankM :: ArrIx P Ix2 Int -> Property
prop_SliceIndexDim2RankM (ArrIx arr' ix@(i :. j)) =
val === (arr <!> (2, i) ! j) .&&.
val === (arr <!> (1, j) ! i)
where
arr = toManifest arr'
val = unsafeIndex arr ix
prop_SliceIndexDim3M :: ArrIx P Ix3 Int -> Property
prop_SliceIndexDim3M (ArrIx arr' ix@(i :> j :. k)) =
val === (arr <! k <! j ! i) .&&.
val === (arr !> i !> j ! k) .&&.
val === (arr <! k !> i ! j) .&&.
val === (arr !> i <! k ! j)
where
arr = toManifest arr'
val = unsafeIndex arr ix
prop_SliceIndexDim3RankM :: ArrIx P Ix3 Int -> Property
prop_SliceIndexDim3RankM (ArrIx arr' ix@(i :> j :. k)) =
val === (arr <!> (3, i) <!> (2, j) ! k) .&&.
val === (arr <!> (3, i) <!> (1, k) ! j) .&&.
val === (arr <!> (2, j) <!> (2, i) ! k) .&&.
val === (arr <!> (2, j) <!> (1, k) ! i) .&&.
val === (arr <!> (1, k) <!> (2, i) ! j) .&&.
val === (arr <!> (1, k) <!> (1, j) ! i)
where
arr = toManifest arr'
val = unsafeIndex arr ix
prop_SliceIndexDim4D :: ArrIx D Ix4 Int -> Property
prop_SliceIndexDim4D (ArrIx arr ix@(i1 :> i2 :> i3 :. i4)) =
val === evaluate' (arr !> i1 !> i2 !> i3) i4 .&&.
val === evaluate' (arr !> i1 !> i2 <! i4) i3 .&&.
val === evaluate' (arr !> i1 <! i4 <! i3) i2 .&&.
val === evaluate' (arr !> i1 <! i4 !> i2) i3 .&&.
val === evaluate' (arr <! i4 !> i1 !> i2) i3 .&&.
val === evaluate' (arr <! i4 !> i1 <! i3) i2 .&&.
val === evaluate' (arr <! i4 <! i3 <! i2) i1 .&&.
val === evaluate' (arr <! i4 <! i3 !> i1) i2
where
val = unsafeIndex arr ix
prop_SliceIndexDim4RankD :: ArrIx D Ix4 Int -> Property
prop_SliceIndexDim4RankD (ArrIx arr ix@(i1 :> i2 :> i3 :. i4)) =
val === unsafeIndex (arr <!> (4, i1) <!> (3, i2) <!> (2, i3)) i4 .&&.
val === unsafeIndex (arr <!> (4, i1) <!> (2, i3) <! i4) i2 .&&.
val === unsafeIndex (arr <!> (3, i2) <!> (3, i1)) (i3 :. i4) .&&.
val === unsafeIndex (arr <!> (2, i3) <!> (2, i2)) (i1 :. i4) .&&.
val === unsafeIndex (arr <!> (2, i3) <!> (1, i4) !> i1) i2 .&&.
val === unsafeIndex (arr <!> (1, i4) !> i1 !> i2) i3
where
val = evaluate' arr ix
prop_SliceIndexDim4RankM :: ArrIx P Ix4 Int -> Property
prop_SliceIndexDim4RankM (ArrIx arr' ix@(i1 :> i2 :> i3 :. i4)) =
val === (arr <!> (4, i1) <!> (3, i2) <!> (2, i3) ! i4) .&&.
val === (arr <!> (4, i1) <!> (2, i3) <! i4 ! i2) .&&.
val === (arr <!> (3, i2) <!> (3, i1) ! (i3 :. i4)) .&&.
val === (arr <!> (2, i3) <!> (2, i2) ! (i1 :. i4)) .&&.
val === (arr <!> (2, i3) <!> (1, i4) !> i1 ! i2) .&&.
val === (arr <!> (1, i4) !> i1 !> i2 ! i3)
where
arr = toManifest arr'
val = unsafeIndex arr ix
prop_SliceIndexDim4M :: ArrIx P Ix4 Int -> Property
prop_SliceIndexDim4M (ArrIx arr' ix@(i1 :> i2 :> i3 :. i4)) =
val === (arr !> i1 !> i2 !> i3 ! i4) .&&.
val === (arr !> i1 !> i2 <! i4 ! i3) .&&.
val === (arr !> i1 <! i4 <! i3 ! i2) .&&.
val === (arr !> i1 <! i4 !> i2 ! i3) .&&.
val === (arr <! i4 !> i1 !> i2 ! i3) .&&.
val === (arr <! i4 !> i1 <! i3 ! i2) .&&.
val === (arr <! i4 <! i3 <! i2 ! i1) .&&.
val === (arr <! i4 <! i3 !> i1 ! i2)
where
arr = toManifest arr'
val = unsafeIndex arr ix
specSliceN :: ( Arbitrary (Array r ix e)
, Show (Array r ix e)
, Slice r ix e
, OuterSlice r ix e
, InnerSlice r ix e
, Eq (Elt r ix e)
, Show (Elt r ix e)
)
=> proxy (r, ix, e) -> Spec
specSliceN proxy =
describe "Slice" $ do
it "SliceRight" $ property $ prop_SliceRight proxy
it "SliceLeft" $ property $ prop_SliceLeft proxy
spec :: Spec
spec = do
describe "Ix1" $
specSizeN (Nothing :: Maybe (D, Ix1, Int))
describe "Ix2" $ do
specSizeN (Nothing :: Maybe (D, Ix2, Int))
specSliceN (Nothing :: Maybe (D, Ix2, Int))
describe "SliceIndex" $ do
it "Delayed" $ property prop_SliceIndexDim2D
it "Rank - Delayed" $ property prop_SliceIndexDim2RankD
it "Manifest" $ property prop_SliceIndexDim2M
it "Rank - Manifest" $ property prop_SliceIndexDim2RankM
describe "Ix3" $ do
specSizeN (Nothing :: Maybe (D, Ix3, Int))
specSliceN (Nothing :: Maybe (D, Ix3, Int))
describe "SliceIndex" $ do
it "Delayed" $ property prop_SliceIndexDim3D
it "Rank - Delayed" $ property prop_SliceIndexDim3RankD
it "Manifest" $ property prop_SliceIndexDim3M
it "Rank - Manifest" $ property prop_SliceIndexDim3RankM
describe "Ix4" $ do
specSizeN (Nothing :: Maybe (D, Ix4, Int))
specSliceN (Nothing :: Maybe (D, Ix4, Int))
describe "SliceIndex" $ do
it "Delayed" $ property prop_SliceIndexDim4D
it "Rank - Delayed" $ property prop_SliceIndexDim4RankD
it "Manifest" $ property prop_SliceIndexDim4M
it "Rank - Manifest" $ property prop_SliceIndexDim4RankM