git.iko.soy/streamly
on master help sign in
Add test for S.groups.
commitbee479ea045863450503a2ea12c55dce2c239052 authorpranaysashank <pranaysashank@composewell.com> dateWed 25 Nov 2020, 15:38 (6 years ago) parentd8a2bba
2 files changed+24−7
compare with master
unifiedsplit
test/Streamly/Test/Prelude/Common.hs+14−5 test/Streamly/Test/Prelude/Serial.hs+10−2
test/Streamly/Test/Prelude/Common.hs+14−5
@@ -10,7 +10,9 @@
1010module Streamly.Test.Prelude.Common
1111 (
1212 -- * Construction operations
13− constructWithReplicate
13+ constructWithRepeat
14+ , constructWithRepeatM
15+ , constructWithReplicate
1416 , constructWithReplicateM
1517 , constructWithIntFromThenTo
1618#if __GLASGOW_HASKELL__ >= 806
@@ -18,10 +20,18 @@ module Streamly.Test.Prelude.Common
1820#endif
1921 , constructWithIterate
2022 , constructWithIterateM
23+ , constructWithEnumerate
24+ , constructWithEnumerateTo
2125 , constructWithFromIndices
2226 , constructWithFromIndicesM
27+ , constructWithFromList
28+ , constructWithFromListM
29+ , constructWithUnfoldr
2330 , constructWithCons
2431 , constructWithConsM
32+ , constructWithYield
33+ , constructWithYieldM
34+ , simpleOps
2535 -- * Applicative operations
2636 , applicativeOps
2737 , applicativeOps1
@@ -196,12 +206,12 @@ constructWithRepeat, constructWithRepeatM
196206constructWithRepeat = constructWithLenM stream list
197207 where
198208 stream n = S.take n $ S.repeat 1
199− list n = return $ take n $ repeat 1
209+ list n = return $ replicate n 1
200210
201211constructWithRepeatM = constructWithLenM stream list
202212 where
203213 stream n = S.take n $ S.repeatM (return 1)
204− list n = return $ take n $ repeat 1
214+ list n = return $ replicate n 1
205215
206216#if __GLASGOW_HASKELL__ >= 806
207217-- XXX try very small steps close to 0
@@ -408,8 +418,7 @@ constructWithYieldM listT op len =
408418 listEquals (==) (listT strm) list
409419
410420simpleProps ::
411− IsStream t
412− => (Int -> t IO Int)
421+ (Int -> t IO Int)
413422 -> (t IO Int -> SerialT IO Int)
414423 -> Int
415424 -> Property
test/Streamly/Test/Prelude/Serial.hs+10−2
@@ -14,7 +14,7 @@ import Control.Monad ( when, forM_ )
1414import Data.Function ( (&) )
1515import Data.IORef ( newIORef, readIORef, writeIORef )
1616import Data.Int (Int64)
17−import Data.List (intercalate)
17+import Data.List (group, intercalate)
1818import Data.Maybe ( isJust, fromJust )
1919import Foreign.Storable (Storable)
2020#if __GLASGOW_HASKELL__ < 808
@@ -287,7 +287,14 @@ testGroupsBy =
287287 (x:_) -> x == minimum ls)
288288 $ S.groupsBy (>) FL.toList
289289 $ S.fromList vec
290− assert $ r == True
290+ assert r
291+
292+testGroups :: Property
293+testGroups =
294+ forAll (choose (0, maxStreamLen)) $ \len ->
295+ forAll (vectorOf len (arbitrary :: Gen Int)) $ \vec -> monadicIO $ do
296+ r <- run $ S.toList $ S.groups FL.toList $ S.fromList vec
297+ assert $ r == group vec
291298
292299-- |
293300-- If the list is empty, returns Nothing,
@@ -562,6 +569,7 @@ main = hspec
562569
563570 describe "Tests for S.groupsBy" $ do
564571 prop "testGroupsBy" testGroupsBy
572+ prop "S.groups = groups" testGroups
565573 prop "testGroupsBySep" testGroupsBySep
566574
567575 describe "Composed MonadThrow serially" $ composeWithMonadThrow serially