From 5c2399d06cfb921f3168ced69a1406746a2a2447 Mon Sep 17 00:00:00 2001 From: rusty1s Date: Sat, 29 Jun 2019 16:18:13 +0200 Subject: [PATCH] test debug data --- test/data/test_data.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/test/data/test_data.py b/test/data/test_data.py index bf3c17f9d7ba..8ff91422deb9 100644 --- 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)