git.iko.soy/streamly
on master help sign in
Create another test group for transformation & add multiple unfolds
e01474badithyaov
test/Streamly/Test/Data/Unfold.hs+42−6
compare with master
unifiedsplit
@@ -20,7 +20,7 @@ import qualified Data.List as List
20
20
21import Control.Monad.Trans.State.Strict
21import Control.Monad.Trans.State.Strict
22import Data.Functor.Identity
22import Data.Functor.Identity
23−import Prelude hiding (const, take, drop, concat)
23+import Prelude hiding (const, take, drop, concat, mapM)
24import Streamly.Prelude (SerialT)
24import Streamly.Prelude (SerialT)
25import Test.Hspec as H
25import Test.Hspec as H
26import Test.Hspec.QuickCheck
26import Test.Hspec.QuickCheck
@@ -228,6 +228,30 @@ enumerateFromToIntegral =
228 let unf = UF.enumerateFromToIntegral t
228 let unf = UF.enumerateFromToIntegral t
229 in testUnfoldD unf f [f .. t]
229 in testUnfoldD unf f [f .. t]
230
230
231+-------------------------------------------------------------------------------
232+-- Stream transformation
233+-------------------------------------------------------------------------------
234+
235+mapM :: Property
236+mapM =
237+ property
238+ $ \f list ->
239+ let fA = apply f :: Int -> Int
240+ fM x = modify (+ 1) >> return (fA x)
241+ unf = UF.mapM fM UF.fromList
242+ mList = Prelude.map fA list
243+ in testUnfoldAD unf 0 (length list) list mList
244+
245+mapMWithInput :: Property
246+mapMWithInput =
247+ property
248+ $ \f list ->
249+ let fA = applyFun2 f :: [Int] -> Int -> Int
250+ fM x y = modify (+ 1) >> return (fA x y)
251+ unf = UF.mapMWithInput fM UF.fromList
252+ mList = Prelude.map (fA list) list
253+ in testUnfoldAD unf 0 (length list) list mList
254+
231take :: Property
255take :: Property
232take =
256take =
233 property
257 property
@@ -359,6 +383,22 @@ testGeneration =
359 -- prop "numFrom" numFrom
383 -- prop "numFrom" numFrom
360 prop "enumerateFromToFractional" enumerateFromToFractional
384 prop "enumerateFromToFractional" enumerateFromToFractional
361
385
386+testTransformation :: Spec
387+testTransformation =
388+ describe "Transformation"
389+ $ do
390+ -- prop "map" map
391+ prop "mapM" mapM
392+ prop "mapMWithInput" mapMWithInput
393+ prop "takeWhileM" takeWhileM
394+ -- prop "takeWhile" takeWhile
395+ prop "take" take
396+ -- prop "filter" filter
397+ prop "filterM" filterM
398+ prop "drop" drop
399+ -- prop "dropWhile" dropWhile
400+ prop "dropWhileM" dropWhileM
401+
362-------------------------------------------------------------------------------
402-------------------------------------------------------------------------------
363-- Main
403-- Main
364-------------------------------------------------------------------------------
404-------------------------------------------------------------------------------
@@ -367,12 +407,8 @@ main :: IO ()
367main = hspec $ do
407main = hspec $ do
368 testInputOps
408 testInputOps
369 testGeneration
409 testGeneration
410+ testTransformation
370 describe "Unfold tests" $ do
411 describe "Unfold tests" $ do
371− prop "take" take
372− prop "takeWhileM" takeWhileM
373− prop "filterM" filterM
374− prop "drop" drop
375− prop "dropWhileM" dropWhileM
376 prop "zipWithM" zipWithM
412 prop "zipWithM" zipWithM
377 prop "concat" concat
413 prop "concat" concat
378 prop "outerProduct" outerProduct
414 prop "outerProduct" outerProduct