From 4c5122e7793c49b80db510f73ecc58ee85e61f8a Mon Sep 17 00:00:00 2001 From: RobinMeersman <77965843+RobinMeersman@users.noreply.github.com> Date: Thu, 7 May 2026 13:57:19 +0200 Subject: [PATCH] Update tests/test_adjacency.py Co-authored-by: Tibo De Peuter --- tests/test_adjacency.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/test_adjacency.py b/tests/test_adjacency.py index e6a1333..27c8b43 100644 --- a/tests/test_adjacency.py +++ b/tests/test_adjacency.py @@ -30,9 +30,13 @@ def test_ring(): assert_symmetric(adj) # each node should connect to itself + 2 neighbors - for i in range(5): - assert adj[i, i] == 1 - assert jnp.sum(adj[i]) == 3 + for node in range(5): + assert adj[node, node] == 1 + assert jnp.sum(adj[node]) == 3 + neighbor1 = (node - 1) % 5 + neighbor2 = (node + 1) % 5 + assert adj[neighbor1, node] == 1 + assert adj[node, neighbor2] == 1 # Symmetrical def test_segment_structure():