From 2f8620c9cefad244f5ab550c652be31d7c690450 Mon Sep 17 00:00:00 2001
From: Jordan Matelsky <j6k4m8@gmail.com>
Date: Sat, 25 Jul 2026 14:06:54 -0400
Subject: [PATCH] test: add real backend performance benchmarks

---
 grand/backends/test_performance.py | 103 +++++++++++++++++++++++++++++
 1 file changed, 103 insertions(+)
 create mode 100644 grand/backends/test_performance.py

diff --git a/grand/backends/test_performance.py b/grand/backends/test_performance.py
new file mode 100644
index 0000000..25facd8
--- /dev/null
+++ b/grand/backends/test_performance.py
@@ -0,0 +1,103 @@
+from contextlib import nullcontext
+
+import pandas as pd
+import pytest
+
+from ._dataframe import DataFrameBackend
+from ._networkx import NetworkXBackend
+
+try:
+    from ._sqlbackend import SQLBackend
+except ImportError:
+    SQLBackend = None
+
+
+pytestmark = pytest.mark.benchmark
+
+
+@pytest.mark.parametrize(
+    "backend",
+    [
+        pytest.param(NetworkXBackend, id="networkx"),
+        pytest.param(DataFrameBackend, id="dataframe"),
+        pytest.param(
+            SQLBackend,
+            marks=pytest.mark.skipif(SQLBackend is None, reason="SQL unavailable"),
+            id="sql",
+        ),
+    ],
+)
+def test_high_degree_neighbor_traversal(backend):
+    kwargs = {"db_url": "sqlite:///:memory:"} if backend is SQLBackend else {}
+    graph = backend(directed=True, **kwargs)
+    transaction = getattr(graph, "transaction", None)
+    with transaction() if transaction else nullcontext():
+        for target in range(500):
+            graph.add_edge("hub", target, {})
+
+    assert len(list(graph.get_node_neighbors("hub"))) == 500
+    if backend is SQLBackend:
+        graph.close()
+
+
+@pytest.mark.skipif(SQLBackend is None, reason="SQL unavailable")
+def test_sql_grouped_transaction_node_insertion():
+    backend = SQLBackend(db_url="sqlite:///:memory:")
+
+    with backend.transaction():
+        for node in range(500):
+            backend.add_node(node, {})
+
+    assert backend.get_node_count() == 500
+    backend.close()
+
+
+@pytest.mark.skipif(SQLBackend is None, reason="SQL unavailable")
+def test_sql_bulk_edge_insertion():
+    backend = SQLBackend(db_url="sqlite:///:memory:", directed=True)
+
+    backend.add_edges_from((node, node + 1) for node in range(500))
+
+    assert backend.get_node_count() == 501
+    assert backend.get_edge_count() == 500
+    backend.close()
+
+
+def test_dataframe_incremental_edge_insertion():
+    backend = DataFrameBackend(directed=True)
+
+    for node in range(250):
+        backend.add_edge(node, node + 1, {})
+
+    assert backend.get_edge_count() == 250
+
+
+@pytest.mark.parametrize(
+    "backend",
+    [
+        pytest.param(NetworkXBackend, id="networkx"),
+        pytest.param(DataFrameBackend, id="dataframe"),
+        pytest.param(
+            SQLBackend,
+            marks=pytest.mark.skipif(SQLBackend is None, reason="SQL unavailable"),
+            id="sql",
+        ),
+    ],
+)
+def test_dataframe_ingestion_with_metadata(backend):
+    kwargs = {"db_url": "sqlite:///:memory:"} if backend is SQLBackend else {}
+    graph = backend(directed=True, **kwargs)
+    edgelist = pd.DataFrame(
+        {
+            "source": range(500),
+            "target": range(1, 501),
+            "weight": range(500),
+            "kind": ["edge"] * 500,
+        }
+    )
+
+    graph.ingest_from_edgelist_dataframe(edgelist, "source", "target")
+
+    assert graph.get_edge_count() == 500
+    if backend is SQLBackend:
+        graph.close()