diff --git a/tests/icosagon/test_convolve.py b/tests/icosagon/test_convolve.py index 5802800..79fb2ac 100644 --- a/tests/icosagon/test_convolve.py +++ b/tests/icosagon/test_convolve.py @@ -188,3 +188,11 @@ def test_dropout_graph_conv_activation_dense_03(): def test_dropout_graph_conv_activation_sparse_03(): _test_dropout_graph_conv_activation_03(True) + + +def test_graph_conv_parameter_count_01(): + adj_mat = torch.rand((10, 20)).round() + + conv = GraphConv(20, 20, adj_mat) + + assert len(list(conv.parameters())) == 1