test debug data

pyg-team · Jun 29, 2019 · 5c2399d · 5c2399d
1 parent e332626
commit 5c2399d
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/test/data/test_data.py b/test/data/test_data.py
@@ -90,3 +90,17 @@ def test_data():
     assert data.num_edge_features == 0
 
     torch_geometric.set_debug(False)
+
+
+def test_debug_data():
+    torch_geometric.set_debug(True)
+
+    Data()
+    Data(edge_index=torch.zeros((2, 0), dtype=torch.long), num_nodes=10)
+    Data(face=torch.zeros((3, 0), dtype=torch.long), num_nodes=10)
+    Data(edge_index=torch.tensor([[0, 1], [1, 0]]), edge_attr=torch.randn(2))
+    Data(x=torch.torch.randn(5, 3), num_nodes=5)
+    Data(pos=torch.torch.randn(5, 3), num_nodes=5)
+    Data(norm=torch.torch.randn(5, 3), num_nodes=5)
+
+    torch_geometric.set_debug(False)