From 23dd68e8b3c17d9c4a149559652b281ba45be1d9 Mon Sep 17 00:00:00 2001
From: Mridul Seth <git@mriduls.com>
Date: Fri, 28 Jul 2023 12:54:50 +0400
Subject: [PATCH] Add basic networkx benchmarks

---
 pyperformance/data-files/benchmarks/MANIFEST  |  1 +
 .../benchmarks/bm_networkx/pyproject.toml     | 12 ++++
 .../benchmarks/bm_networkx/requirements.txt   |  1 +
 .../benchmarks/bm_networkx/run_benchmark.py   | 63 +++++++++++++++++++
 4 files changed, 77 insertions(+)
 create mode 100644 pyperformance/data-files/benchmarks/bm_networkx/pyproject.toml
 create mode 100644 pyperformance/data-files/benchmarks/bm_networkx/requirements.txt
 create mode 100644 pyperformance/data-files/benchmarks/bm_networkx/run_benchmark.py

diff --git a/pyperformance/data-files/benchmarks/MANIFEST b/pyperformance/data-files/benchmarks/MANIFEST
index 4420f98b..e9869ea2 100644
--- a/pyperformance/data-files/benchmarks/MANIFEST
+++ b/pyperformance/data-files/benchmarks/MANIFEST
@@ -53,6 +53,7 @@ mako	<local>
 mdp	<local>
 meteor_contest	<local>
 nbody	<local>
+networkx	<local>
 nqueens	<local>
 pathlib	<local>
 pickle	<local>
diff --git a/pyperformance/data-files/benchmarks/bm_networkx/pyproject.toml b/pyperformance/data-files/benchmarks/bm_networkx/pyproject.toml
new file mode 100644
index 00000000..c81909e5
--- /dev/null
+++ b/pyperformance/data-files/benchmarks/bm_networkx/pyproject.toml
@@ -0,0 +1,12 @@
+[project]
+name = "pyperformance_bm_networkx"
+requires-python = ">=3.8"
+dependencies = [
+    "pyperf",
+    "networkx",
+]
+urls = {repository = "https://github.com/python/pyperformance"}
+dynamic = ["version"]
+
+[tool.pyperformance]
+name = "networkx"
diff --git a/pyperformance/data-files/benchmarks/bm_networkx/requirements.txt b/pyperformance/data-files/benchmarks/bm_networkx/requirements.txt
new file mode 100644
index 00000000..2bd69c8d
--- /dev/null
+++ b/pyperformance/data-files/benchmarks/bm_networkx/requirements.txt
@@ -0,0 +1 @@
+networkx==3.1
diff --git a/pyperformance/data-files/benchmarks/bm_networkx/run_benchmark.py b/pyperformance/data-files/benchmarks/bm_networkx/run_benchmark.py
new file mode 100644
index 00000000..70f823f9
--- /dev/null
+++ b/pyperformance/data-files/benchmarks/bm_networkx/run_benchmark.py
@@ -0,0 +1,63 @@
+"""
+Benchmark graph creation and algorithms from NetworkX.
+"""
+
+import pyperf
+
+import networkx as nx
+
+
+G = nx.erdos_renyi_graph(200, 0.02)
+
+
+def bench_sparse_graph_creation():
+    _ = nx.erdos_renyi_graph(100, 0.02)
+
+
+def bench_dense_graph_creation():
+    _ = nx.erdos_renyi_graph(100, 0.9)
+
+
+def bench_betweenness():
+    _ = nx.betweenness_centrality(G)
+
+
+def bench_modularity():
+    _ = nx.community.greedy_modularity_communities(G)
+
+
+BENCHMARKS = (
+    "sparse_graph_creation",
+    "dense_graph_creation",
+    "betweenness",
+    "modularity",
+)
+
+
+def bench_networkx(loops, func):
+    t0 = pyperf.perf_counter()
+    for _ in range(loops):
+        func()
+    return pyperf.perf_counter() - t0
+
+
+def add_cmdline_args(cmd, args):
+    if args.benchmark:
+        cmd.append(args.benchmark)
+
+
+if __name__ == "__main__":
+    runner = pyperf.Runner(add_cmdline_args=add_cmdline_args)
+    runner.metadata["description"] = "NetworkX benchmark"
+    runner.argparser.add_argument("benchmark", nargs="?", choices=sorted(BENCHMARKS))
+
+    args = runner.parse_args()
+    if args.benchmark:
+        benchmarks = (args.benchmark,)
+    else:
+        benchmarks = sorted(BENCHMARKS)
+
+    for bench in benchmarks:
+        name = "networkx_%s" % bench
+        func = globals()["bench_" + bench]
+        runner.bench_time_func(name, bench_networkx, func)