git.iko.soy/streamly
on master help sign in
Create another test group for combination & add concatMapM
de680fbadithyaov
test/Streamly/Test/Data/Unfold.hs+44−8
compare with master
unifiedsplit
@@ -308,6 +308,10 @@ dropWhileM =
308 fS = Prelude.length n - Prelude.length fL
308 fS = Prelude.length n - Prelude.length fL
309 in testUnfoldAD unf 0 fS n fL
309 in testUnfoldAD unf 0 fS n fL
310
310
311+-------------------------------------------------------------------------------
312+-- Stream combination
313+-------------------------------------------------------------------------------
314+
311zipWithM :: Property
315zipWithM :: Property
312zipWithM =
316zipWithM =
313 property
317 property
@@ -336,6 +340,22 @@ outerProduct =
336 lst = [(a, b) :: (Int, Int) | a <- [0 .. 10], b <- [0 .. 20]]
340 lst = [(a, b) :: (Int, Int) | a <- [0 .. 10], b <- [0 .. 20]]
337 in testUnfold unf ((0, 0) :: (Int, Int)) lst
341 in testUnfold unf ((0, 0) :: (Int, Int)) lst
338
342
343+concatMapM :: Bool
344+concatMapM =
345+ let unfInF b =
346+ modify (+ 1)
347+ >> return
348+ (UF.lmap
349+ (\() -> undefined)
350+ (UF.supply
351+ (UF.replicateM 10)
352+ (modify (+ 1) >> return b)))
353+ listInF b = replicate 10 b
354+ unfOut = UF.enumerateFromToIntegral 10
355+ unf = UF.concatMapM unfInF unfOut
356+ list = List.concatMap listInF [1 .. 10]
357+ in testUnfoldAD unf 0 110 1 list
358+
339-------------------------------------------------------------------------------
359-------------------------------------------------------------------------------
340-- Test groups
360-- Test groups
341-------------------------------------------------------------------------------
361-------------------------------------------------------------------------------
@@ -399,19 +419,35 @@ testTransformation =
399 -- prop "dropWhile" dropWhile
419 -- prop "dropWhile" dropWhile
400 prop "dropWhileM" dropWhileM
420 prop "dropWhileM" dropWhileM
401
421
422+
423+testCombination :: Spec
424+testCombination =
425+ describe "Transformation"
426+ $ do
427+ prop "zipWithM" zipWithM
428+ -- prop "zipWith" zipWith
429+ -- prop "teeZipWith" teeZipWith
430+ prop "concat" concat
431+ prop "concatMapM" concatMapM
432+ prop "outerProduct" outerProduct
433+ -- prop "ap" ap
434+ -- prop "apDiscardFst" apDiscardFst
435+ -- prop "apDiscardSnd" apDiscardSnd
436+
437+
402-------------------------------------------------------------------------------
438-------------------------------------------------------------------------------
403-- Main
439-- Main
404-------------------------------------------------------------------------------
440-------------------------------------------------------------------------------
405
441
406main :: IO ()
442main :: IO ()
443+main =
407−main = hspec $ do
444+ hspec
408− testInputOps
409− testGeneration
410− testTransformation
411− describe "Unfold tests" $ do
445+ $ describe "Unfold tests"
412− prop "zipWithM" zipWithM
446+ $ do
413− prop "concat" concat
447+ testInputOps
414− prop "outerProduct" outerProduct
448+ testGeneration
449+ testTransformation
450+ testCombination
415 -- prop "concatMapM" concatMapM
451 -- prop "concatMapM" concatMapM
416 -- prop "gbracket" gbracket
452 -- prop "gbracket" gbracket
417 -- prop "gbracketIO" gbracketIO
453 -- prop "gbracketIO" gbracketIO