Add set operations-like benchmark for Data.Map.partitionKeys

containers-tests/benchmarks/SetOperations/PartitionKeys-MapSet.hs

@@ -0,0 +1,14 @@
+module Main where
+
+import qualified Data.Map as M
+import qualified Data.Set as S
+import SetOperations
+import Utils.Containers.Internal.StrictPair
+
+main =
+  benchmark'
+    (\xs -> M.fromList [(x, x) | x <- xs])
+    S.fromList
+    True
+    [ ("partitionKeys", M.partitionKeys)
+    ]

containers-tests/containers-tests.cabal

@@ -227,6 +227,15 @@ benchmark set-operations-set
   other-modules:  SetOperations
   ghc-options:    -O2
 
+benchmark partition-keys-mapset
+  import: benchmark-deps
+  default-language: Haskell2010
+  type:           exitcode-stdio-1.0
+  hs-source-dirs: benchmarks/SetOperations
+  main-is:        PartitionKeys-MapSet.hs
+  other-modules:  SetOperations
+  ghc-options:    -O2
+
 benchmark lookupge-intmap
   import: benchmark-deps
   default-language: Haskell2010