(self)
| 36 | class ConvTest(tf.test.TestCase): |
| 37 | |
| 38 | def testConv(self): |
| 39 | edge_index = tf.constant([[0, 1, 1, 2], [1, 0, 2, 1]], dtype=tf.int32) |
| 40 | x = tf.constant([[-1], [0], [1]], dtype=tf.float32) |
| 41 | x = [x, x] |
| 42 | conv = SimpleConv() |
| 43 | x1 = conv(x, edge_index, size=[3, 3]) |
| 44 | |
| 45 | with self.test_session(): |
| 46 | self.assertAllEqual([[0.], [0.], [0.]], x1.eval()) |
| 47 | |
| 48 | |
| 49 | if __name__ == '__main__': |