Add more benchmarks

Author: Peter554

tests/benchmarking/test_benchmarking.py

@@ -68,6 +68,26 @@ def test_build_django_from_cache(benchmark):
         benchmark(fn)
 
 
+def test_find_descendants(large_graph, benchmark):
+    result = benchmark(large_graph.find_descendants, "mypackage")
+    assert len(result) == 28222
+
+
+def test_find_downstream_modules(large_graph, benchmark):
+    result = benchmark(large_graph.find_downstream_modules, DEEP_LAYERS[0], as_package=True)
+    assert len(result) == 80
+
+
+def test_find_upstream_modules(large_graph, benchmark):
+    result = benchmark(large_graph.find_upstream_modules, DEEP_LAYERS[0], as_package=True)
+    assert len(result) == 2159
+
+
+def test_find_shortest_chain(large_graph, benchmark):
+    result = benchmark(large_graph.find_shortest_chain, DEEP_LAYERS[0], DEEP_LAYERS[1])
+    assert result is not None and len(result) == 5
+
+
 def test_find_shortest_chain__no_chain(large_graph, benchmark):
     result = benchmark(
         large_graph.find_shortest_chain,