Add benchmark for iterating over modules in graph

We do this currently in import-linter remove_ignored_imports.
seddonym · Feb 6, 2025 · f99d32d · f99d32d
1 parent 3510cb4
commit f99d32d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/benchmarking/test_benchmarking.py b/tests/benchmarking/test_benchmarking.py
@@ -430,3 +430,11 @@ def f(n):
             _ = f"foo{i}" in large_graph.modules
 
     _run_benchmark(benchmark, f, 100)
+
+
+def test_iterate_over_modules_in_graph(large_graph, benchmark):
+    def f():
+        for module in large_graph.modules:
+            _ = module
+
+    _run_benchmark(benchmark, f)