uds/test.icl

   1 module test
   2
   3 import StdEnv
   4 import Data.Either
   5 import Data.Func
   6 from Data.Map import :: Map(..)
   7 import qualified Data.Map
   8 import Control.Monad
   9 import Control.Monad.State
  10 import Control.Monad.Fail
  11
  12 import ASDS
  13 import ASDS.Source
  14 import ASDS.Lens
  15 import ASDS.Select
  16 import ASDS.Parallel
  17
  18 instance MonadFail (Either String) where fail s = Left s
  19
  20
  21 readwrite :: r w (sds m () r w) -> m () | MonadFail m & read sds & write sds & TC r & TC w & == r
  22 readwrite r w sds = equal r (setShare w sds >>| getShare sds)
  23
  24 equal :: a (m a) -> m () | MonadFail m & == a
  25 equal expected mon = mon >>= \v->if (v == expected) (pure ()) (fail "Not equal")
  26
  27 Start :: Either String ((), Map String Dynamic)
  28 Start = runStateT (sequence_ $ map test tests) 'Data.Map'.newMap
  29 where
  30         test t = put 'Data.Map'.newMap >>| t
  31
  32         tests = flatten
  33                 [ [readwrite i i sh \\ i<-[-1, 0, -100, 100]]
  34                 , [setShare [0..10] sh >>| equal i (getShare (focus ((), i) (indexedStore sh)))\\i<-[0..10]]
  35                 , [setShare [0..10] sh >>| setShare 42 (focus ((), i) (indexedStore sh)) >>| equal 42 (getShare (focus ((), i) (indexedStore sh)))\\i<-[0..10]]
  36                 , [setShare [(i, i)\\i<-[0..10]] sh >>| equal i (getShare (focus ((), i) (assocStore sh)))\\i<-[0..10]]
  37                 , [setShare ('Data.Map'.fromList [(i, i)\\i<-[0..10]]) sh >>| equal i (getShare (focus ((), i) (keyedStore sh)))\\i<-[0..10]]
  38                 , [setShare [0..10] sh >>| setShare 4 (focus "idx" astore) >>| equal 4 (getShare $ focus ((), ()) $ selectList (focus "idx" astore) sh)]
  39                 , [setShare 38 sh >>| equal 38 (getShare (After 100 sh (pure ())))]
  40                 , [testpar (focus "foo" astore) (focus "bar" astore)]
  41                 , [testpar (after 100 $ focus "foo" astore) (after 100 $ focus "bar" astore)]
  42                 , [testpar (after 100 $ focus "foo" astore) (focus "bar" astore)]
  43                 , [testpar (focus "foo" astore) (after 100 $ focus "bar" astore)]
  44 //selectList :: (sdss m p1 Int w1) (sdsc m p2 [a] [a]) -> Select sdss (Lens sdsc) m (p1, p2) a a | TC p1 & TC p2 & TC a & TC w1 & MonadFail m
  45                 ]
  46
  47         sh :: Lens (Lens (Lens (Lens (RWPair ReadSource WriteSource)))) (StateT (Map String Dynamic) m) () a a | MonadFail m & TC a
  48         sh = focus "foo" astore
  49
  50 testpar :: (A.a: sds1 m () a a | TC, == a) (A.a: sds2 m () a a | TC, == a) -> m () | MonadFail m & read, write sds1 & read, write sds2
  51 testpar l r =
  52             setShare (42, 'a') (l >+< r)
  53         >>| equal (42, 'a') (getShare $ l >+< r) >>| equal 42 (getShare l) >>| equal 'a' (getShare r)
  54         >>| setShare 38 l
  55         >>| equal (38, 'a') (getShare $ l >+< r) >>| equal 38 (getShare l) >>| equal 'a' (getShare r)
  56         >>| setShare 'b' r
  57         >>| equal (38, 'b') (getShare $ l >+< r) >>| equal 38 (getShare l) >>| equal 'b' (getShare r)
  58
  59 :: After sds m p r w = After Int (sds m p r w) (m ())
  60 after :: Int (sds m p r w) -> After sds m p r w | pure m
  61 after i sds = After i sds $ pure ()
  62
  63 instance read (After sds) | read sds
  64 where
  65         read (After 0 sds _) p = read sds p
  66         read (After n sds m) p = pure (Reading (After (n-1) sds m))
  67
  68 instance write (After sds) | write sds
  69 where
  70         write (After 0 sds _) p w = write sds p w
  71         write (After n sds m) p w = pure (Writing (After (n-1) sds m))
  72
  73 astore :: Lens (Lens (Lens (RWPair ReadSource WriteSource))) (StateT (Map String Dynamic) m) String a a | MonadFail m & TC a
  74 astore = fromDynStore dstore
  75
  76 dstore :: Lens (Lens (RWPair ReadSource WriteSource)) (StateT (Map String Dynamic) m) String Dynamic Dynamic | MonadFail m
  77 dstore = translate (\i->((), i)) $ keyedStore store
  78
  79 store :: Source (StateT (Map String Dynamic) m) () (Map String Dynamic) (Map String Dynamic) | Monad m
  80 store = source (\_->getState) \_->put