diff options
Diffstat (limited to 'test/Main.hs')
| -rw-r--r-- | test/Main.hs | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/test/Main.hs b/test/Main.hs index d29e4d5..0a07531 100644 --- a/test/Main.hs +++ b/test/Main.hs @@ -7,17 +7,17 @@ import Data.Array.Nested arr :: Ranked I2 (Shaped [2, 3] (Double, Int)) -arr = rgenerate (3 ::: 4 ::: IZR) $ \(i ::: j ::: IZR) -> - sgenerate @[2, 3] $ \(k ::$ l ::$ IZS) -> +arr = rgenerate (3 :.: 4 :.: ZIR) $ \(i :.: j :.: ZIR) -> + sgenerate @[2, 3] $ \(k :.$ l :.$ ZIS) -> let s = 24*i + 6*j + 3*k + l in (fromIntegral s, s) foo :: (Double, Int) -foo = arr `rindex` (2 ::: 1 ::: IZR) `sindex` (1 ::$ 1 ::$ IZS) +foo = arr `rindex` (2 :.: 1 :.: ZIR) `sindex` (1 :.$ 1 :.$ ZIS) bad :: Ranked I2 (Ranked I1 Double) -bad = rgenerate (3 ::: 4 ::: IZR) $ \(i ::: j ::: IZR) -> - rgenerate (i ::: IZR) $ \(k ::: IZR) -> +bad = rgenerate (3 :.: 4 :.: ZIR) $ \(i :.: j :.: ZIR) -> + rgenerate (i :.: ZIR) $ \(k :.: ZIR) -> let s = 24*i + 6*j + 3*k in fromIntegral s |