def test_connect(self): n_inputs = 10 n_hidden = 20 x = tf.ones([1, n_inputs]) input_layer = Layer(n_units=n_inputs,activation=x) id_layer1 = Layer(n_hidden,activation=tf.identity) id_layer2 = Layer(n_hidden,activation=tf.identity) nn = NeuralNetwork() # layer 0 nn.add_layer(input_layer) # layer 1 nn.add_layer(id_layer1) # layer 2 nn.add_layer(id_layer2) self.assertEqual(nn.size(),3) # connect input to output self.assertEqual(nn.graph.number_of_edges(),2) nn.connect_layers(0, 2) self.assertEqual(nn.graph.number_of_edges(),3) self.assertIsNot(nn.weights(0,1),nn.weights(0,2)) self.assertIsNot(nn.weights(0,1),nn.weights(1,2)) self.assertIsNot(nn.weights(1,2),nn.weights(0,2))