Add EdgeIndex identity tests (#9238)

pyg-team · Apr 25, 2024 · 3573d21 · 3573d21
1 parent 5c4aa64
commit 3573d21
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/test_edge_index.py b/test/test_edge_index.py
@@ -96,6 +96,10 @@ def test_identity(dtype, device, is_undirected):
     assert out.sort_order == adj.sort_order
     assert out.is_undirected == adj.is_undirected
 
+    out = EdgeIndex(adj, sparse_size=(4, 4), sort_order='row')
+    assert out.sparse_size() == (4, 4)
+    assert out.sort_order == 'row'
+
 
 @withCUDA
 @pytest.mark.parametrize('dtype', DTYPES)