From 108140b45e1ad86f9d17ca4ef50e40fa0aaa6165 Mon Sep 17 00:00:00 2001 From: Stanislaw Adaszewski Date: Sat, 20 Jun 2020 14:53:26 +0200 Subject: [PATCH] Add test_graph_conv_parameter_count_01(). --- tests/icosagon/test_convolve.py | 8 ++++++++ 1 file changed, 8 insertions(+) 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