Add benchmarks

Author: rnjtranjan

benchmark/Streamly/Benchmark/Prelude/Serial/NestedStream.hs

@@ -436,12 +436,30 @@ joinInnerMap val1 val2 _ =
             (fmap toKvMap (mkStreamLen val1))
             (fmap toKvMap (mkStreamLen val2))
 
+{-# INLINE intersectBy #-}
+intersectBy :: Int -> Int -> Int -> IO ()
+intersectBy val1 val2 _ =
+        S.drain $
+            Internal.intersectBy (==)
+            (fmap toKvMap (mkStreamLen val1))
+            (fmap toKvMap (mkStreamLen val2))
+
+{-# INLINE intersectBySorted #-}
+intersectBySorted :: Int -> Int -> Int -> IO ()
+intersectBySorted val1 val2 _ =
+        S.drain $
+            Internal.intersectBySorted compare
+            (fmap toKvMap (mkStreamLen val1))
+            (fmap toKvMap (mkStreamLen val2))
+
 o_n_heap_buffering :: Int -> [Benchmark]
 o_n_heap_buffering value =
     [ bgroup "buffered"
         [
           benchIOSrc1 "joinInner" (joinInner sqrtVal sqrtVal)
         , benchIOSrc1 "joinInnerMap" (joinInnerMap sqrtVal sqrtVal)
+        , benchIOSrc1 "intersectBy" (intersectBy sqrtVal sqrtVal)
+        , benchIOSrc1 "intersectBySorted" (intersectBySorted sqrtVal sqrtVal)
         ]
     ]
 

test/Streamly/Test/Prelude/Top.hs

@@ -66,6 +66,25 @@ joinInnerMap =
                           ]
                 assert (v1 == v2)
 
+intersectBy :: Property
+intersectBy =
+    forAll (listOf (chooseInt (min_value, max_value))) $ \ls0 ->
+        forAll (listOf (chooseInt (min_value, max_value))) $ \ls1 ->
+            monadicIO $ action (sort ls0) (sort ls1)
+
+            where
+
+            action ls0 ls1 = do
+                v1 <-
+                    run
+                    $ S.toList
+                    $ Top.intersectBy
+                        (==)
+                        (S.fromList ls0)
+                        (S.fromList ls1)
+                let v2 = intersect ls0 ls1
+                assert (v1 == sort v2)
+
 intersectBySorted :: Property
 intersectBySorted =
     forAll (listOf (chooseInt (min_value, max_value))) $ \ls0 ->
@@ -97,4 +116,5 @@ main = hspec $ do
         prop "joinInner" Main.joinInner
         prop "joinInnerMap" Main.joinInnerMap
         -- intersect
+        prop "intersectBy" Main.intersectBy
         prop "intersectBySorted" Main.intersectBySorted