git.iko.soy/streamly
on master help sign in
Add a test case for Fold.many
7950a0aAdithya Kumar
test/Streamly/Test/Data/Fold.hs+16−2
compare with master
unifiedsplit
@@ -18,7 +18,7 @@ import Test.QuickCheck.Monadic (monadicIO, assert, run)
18
18
19import qualified Prelude
19import qualified Prelude
20import qualified Streamly.Internal.Data.Fold as F
20import qualified Streamly.Internal.Data.Fold as F
21−import qualified Streamly.Prelude as S
21+import qualified Streamly.Internal.Data.Stream.IsStream as S
22import qualified Streamly.Data.Fold as FL
22import qualified Streamly.Data.Fold as FL
23
23
24import Prelude hiding
24import Prelude hiding
@@ -417,6 +417,20 @@ unzip =
417 let v2 = (6, ["aa", "bb", "cc"])
417 let v2 = (6, ["aa", "bb", "cc"])
418 assert (v1 == v2)
418 assert (v1 == v2)
419
419
420+many :: Property
421+many =
422+ forAll (listOf (chooseInt (0, 100))) $ \lst ->
423+ forAll (chooseInt (1, 100)) $ \i ->
424+ monadicIO $ do
425+ let strm = S.fromList lst
426+ r1 <- S.fold (F.many (split i) F.toList) strm
427+ r2 <- S.toList $ S.foldMany (split i) strm
428+ assert $ r1 == r2
429+
430+ where
431+
432+ split i = F.take i F.toList
433+
420main :: IO ()
434main :: IO ()
421main = hspec $
435main = hspec $
422 describe "Fold" $ do
436 describe "Fold" $ do
@@ -493,6 +507,6 @@ main = hspec $
493 prop "unzip" Main.unzip
507 prop "unzip" Main.unzip
494
508
495 -- Nesting
509 -- Nesting
496− -- many
510+ prop "many" Main.many
497 -- concatMap
511 -- concatMap
498 -- chunksOf
512 -- chunksOf