test: add more-itertools to benchmarks

benchmarks/__init__.py

@@ -2,31 +2,47 @@
 
 from .scenarios import (
     benchmark_itertools_cycle,
+    benchmark_more_itertools_filter_map,
+    benchmark_more_itertools_flatten,
+    benchmark_more_itertools_nth,
+    benchmark_more_itertools_take,
+    benchmark_more_itertools_unzip,
     benchmark_rusty_iterators_cache_cycle,
     benchmark_rusty_iterators_copy_cycle,
     benchmark_rusty_iterators_count,
     benchmark_rusty_iterators_filter,
+    benchmark_rusty_iterators_filter_map,
     benchmark_rusty_iterators_flatten,
     benchmark_rusty_iterators_map,
+    benchmark_rusty_iterators_nth,
     benchmark_rusty_iterators_sequence_wrapper,
+    benchmark_rusty_iterators_take,
+    benchmark_rusty_iterators_unzip,
     benchmark_stdlib_count,
     benchmark_stdlib_filter,
-    benchmark_stdlib_flatten,
     benchmark_stdlib_map,
 )
 
 __all__ = (
     "benchmark_itertools_cycle",
+    "benchmark_more_itertools_filter_map",
+    "benchmark_more_itertools_flatten",
+    "benchmark_more_itertools_nth",
+    "benchmark_more_itertools_take",
+    "benchmark_more_itertools_unzip",
     "benchmark_rusty_iterators_cache_cycle",
     "benchmark_rusty_iterators_copy_cycle",
     "benchmark_rusty_iterators_count",
     "benchmark_rusty_iterators_filter",
+    "benchmark_rusty_iterators_filter_map",
     "benchmark_rusty_iterators_flatten",
     "benchmark_rusty_iterators_map",
+    "benchmark_rusty_iterators_nth",
     "benchmark_rusty_iterators_sequence_wrapper",
+    "benchmark_rusty_iterators_take",
+    "benchmark_rusty_iterators_unzip",
     "benchmark_stdlib_count",
     "benchmark_stdlib_filter",
-    "benchmark_stdlib_flatten",
     "benchmark_stdlib_map",
 )
 

benchmarks/scenarios.py

@@ -3,6 +3,8 @@
 from itertools import cycle
 from typing import TYPE_CHECKING
 
+import more_itertools as mi
+
 from rusty_iterators import LIter
 
 from .manager import BenchmarkManager
@@ -58,8 +60,8 @@ def benchmark_rusty_iterators_flatten(arg: Iterable[list[int]]) -> None:
 
 
 @BenchmarkManager.register(arg=[[i, i * 2] for i in range(1_000_000)])
-def benchmark_stdlib_flatten(arg: Iterable[list[int]]) -> None:
-    _ = [el for sub in arg for el in sub]
+def benchmark_more_itertools_flatten(arg: Iterable[list[int]]) -> None:
+    _ = list(mi.flatten(arg))
 
 
 @BenchmarkManager.register(arg=range(1_000_000))
@@ -70,3 +72,45 @@ def benchmark_rusty_iterators_count(arg: Iterable[int]) -> None:
 @BenchmarkManager.register(arg=range(1_000_000))
 def benchmark_stdlib_count(arg: Iterable[int]) -> None:
     _ = len(list(arg))
+
+
+@BenchmarkManager.register(arg=[[i, i * 2] for i in range(1_000_000)])
+def benchmark_more_itertools_unzip(arg: Iterable[list[int]]) -> None:
+    left, right = mi.unzip(arg)
+    list(left)
+    list(right)
+
+
+@BenchmarkManager.register(arg=[[i, i * 2] for i in range(1_000_000)])
+def benchmark_rusty_iterators_unzip(arg: Iterable[list[int]]) -> None:
+    _left, _right = LIter.from_items(*arg).unzip()
+
+
+@BenchmarkManager.register(arg=range(1_000_000))
+def benchmark_more_itertools_filter_map(arg: Iterable[int]) -> None:
+    _ = list(mi.filter_map(lambda x: x**2 if x % 2 == 0 else None, arg))
+
+
+@BenchmarkManager.register(arg=range(1_000_000))
+def benchmark_rusty_iterators_filter_map(arg: Iterable[int]) -> None:
+    _ = LIter.from_it(iter(arg)).filter(lambda x: x % 2 == 0).map(lambda x: x**2).collect()
+
+
+@BenchmarkManager.register(arg=range(1_000_000))
+def benchmark_more_itertools_nth(arg: Iterable[int]) -> None:
+    _ = mi.nth(arg, 999_999)
+
+
+@BenchmarkManager.register(arg=range(1_000_000))
+def benchmark_rusty_iterators_nth(arg: Iterable[int]) -> None:
+    _ = LIter.from_it(iter(arg)).nth(999_999)
+
+
+@BenchmarkManager.register(arg=range(1_000_000))
+def benchmark_more_itertools_take(arg: Iterable[int]) -> None:
+    _ = list(mi.take(500_000, arg))
+
+
+@BenchmarkManager.register(arg=range(1_000_000))
+def benchmark_rusty_iterators_take(arg: Iterable[int]) -> None:
+    _ = LIter.from_it(iter(arg)).take(500_000).collect()

pyproject.toml

@@ -22,6 +22,11 @@ dev = [
     "cython==3.0.11",
     "types-setuptools==75.8.0.20250110",
 ]
+
+benchmarks = [
+    "more-itertools==10.6.0",
+]
+
 docs = ["mkdocs==1.6.1"]
 all = [{ include-group = "dev" }, { include-group = "docs" }]