Update tests/test_adjacency.py
Co-authored-by: Tibo De Peuter <tibo.depeuter@ugent.be>
This commit is contained in:
parent
7026125bba
commit
4c5122e779
1 changed files with 7 additions and 3 deletions
|
|
@ -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():
|
||||
|
|
|
|||
Reference in a new issue