Merge pull request #180 from Peter554/add-benchmark.module-in-graph-m…

…odules

Add benchmark for "..." in graph.modules

tests/benchmarking/test_benchmarking.py

@@ -44,7 +44,6 @@ def large_graph():
     f"{DEEP_PACKAGE}.application.3242334296.2454157946",
 )
 
-
 TOP_LEVEL_PACKAGE_DEPENDENCIES = {
     PackageDependency(
         importer="mypackage.domain",
@@ -423,3 +422,19 @@ def test_no_chains(self, large_graph, benchmark):
 
 def test_copy_graph(large_graph, benchmark):
     _run_benchmark(benchmark, lambda: deepcopy(large_graph))
+
+
+def test_graph_contains_module(large_graph, benchmark):
+    def f(n):
+        for i in range(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)