1ab2878 · streamly · git.iko.soy

core/src/Streamly/Data/Stream.hs+1

compare with master

unifiedsplit

@@ -431,6 +431,7 @@ module Streamly.Data.Stream

431 -- >>> wordsBy p f = Stream.parseMany (Parser.wordBy p f)

432 -- >>> groupsOf n f = Stream.foldMany (Fold.take n f)

433 , foldMany -- XXX Rename to foldRepeat

434+ , groupsOf

434 , parseMany

435 , parseMany

435 , Array.chunksOf

436 , Array.chunksOf

436

437

core/src/Streamly/Internal/Data/Stream/StreamD/Type.hs+11

@@ -1903,6 +1903,17 @@ foldMany (Fold fstep initial extract) (Stream step state) =

1903 step' _ (FoldManyYield b next) = return $ Yield b next

1904 step' _ FoldManyDone = return Stop

1905

1906+-- | Group the input stream into groups of @n@ elements each and then fold each

1907+-- group using the provided fold function.

1908+--

1909+-- @groupsOf n f = foldMany (FL.take n f)@

1910+--

1911+-- >>> Stream.toList $ Stream.groupsOf 2 Fold.sum (Stream.enumerateFromTo 1 10)

1912+-- [3,7,11,15,19]

1913+--

1914+-- This can be considered as an n-fold version of 'take' where we apply

1915+-- 'take' repeatedly on the leftover stream until the stream exhausts.

1916+--

1906{-# INLINE groupsOf #-}

1917{-# INLINE groupsOf #-}

1907groupsOf :: Monad m => Int -> Fold m a b -> Stream m a -> Stream m b

1918groupsOf :: Monad m => Int -> Fold m a b -> Stream m a -> Stream m b

1908groupsOf n f = foldMany (FL.take n f)

1919groupsOf n f = foldMany (FL.take n f)

hie.yaml+2

@@ -94,6 +94,8 @@ cradle:

94 cabal:

95 - path: "./test/Streamly/Test/Data/Array.hs"

96 component: "test:Data.Array"

97+ - path: "./test/Streamly/Test/Data/Stream.hs"

98+ component: "test:Data.Stream"

97 - path: "./test/Streamly/Test/Data/Array/Generic.hs"

99 - path: "./test/Streamly/Test/Data/Array/Generic.hs"

98 component: "test:Data.Array.Generic"

100 component: "test:Data.Array.Generic"

99 - path: "./test/Streamly/Test/Data/Array/Mut.hs"

101 - path: "./test/Streamly/Test/Data/Array/Mut.hs"

test/Streamly/Test/Data/Stream.hsnew file+33

@@ -0,0 +1,33 @@

1+-- |

2+-- Module : Streamly.Test.Data.Stream

4+-- License : BSD-3-Clause

5+-- Maintainer : streamly@composewell.com

6+-- Stability : experimental

7+-- Portability : GHC

8+

9+module Streamly.Test.Data.Stream (main) where

10+

11+import qualified Streamly.Data.Fold as Fold

12+import qualified Streamly.Data.Stream as Stream

13+

14+import Test.Hspec.QuickCheck

15+import Test.Hspec as H

16+

17+moduleName :: String

18+moduleName = "Data.Stream"

19+

20+testgroupsOf :: Expectation

21+testgroupsOf =

22+ Stream.toList

23+ (Stream.groupsOf 2 Fold.sum (Stream.enumerateFromTo 1 10))

24+ `shouldReturn` [3::Int, 7, 11, 15, 19]

25+

26+main :: IO ()

27+main =

28+ hspec $

29+ H.parallel $

30+ modifyMaxSuccess (const 1) $ do

31+ describe moduleName $ do

32+ describe "Repeated Fold" $ do

33+ prop "testgroupsOf" testgroupsOf

test/streamly-tests.cabal+6

@@ -241,6 +241,12 @@ test-suite Data.Array

241 main-is: Streamly/Test/Data/Array.hs

242 ghc-options: -main-is Streamly.Test.Data.Array.main

243

244+test-suite Data.Stream

245+ import: test-options

246+ type: exitcode-stdio-1.0

247+ main-is: Streamly/Test/Data/Stream.hs

248+ ghc-options: -main-is Streamly.Test.Data.Stream.main

249+

244test-suite Data.Array.Mut

250test-suite Data.Array.Mut

245 import: test-options

251 import: test-options

246 type: exitcode-stdio-1.0

252 type: exitcode-stdio-1.0