def test_add_node(self): layer = Layer(0, 'input') layer.default_activation_type = 'linear' node = Node() layer.add_node(node) self.assertEqual(1, layer.total_nodes()) self.assertEqual(0, layer.nodes[0].node_no) self.assertEqual('linear', layer.nodes[0].get_activation_type()) layer.default_activation_type = 'sigmoid' node = Node() layer.add_node(node) self.assertEqual(2, layer.total_nodes()) self.assertEqual(1, layer.nodes[1].node_no) self.assertEqual('sigmoid', layer.nodes[1].get_activation_type()) node = BiasNode() layer.add_node(node) self.assertEqual(3, layer.total_nodes()) self.assertEqual(2, layer.nodes[2].node_no) node = Node() node.set_activation_type('tanh') layer.add_node(node) self.assertEqual('tanh', layer.nodes[3].get_activation_type())
def test_total_nodes(self): layer = Layer(0, 'input') layer.add_nodes(2, 'input') layer.add_nodes(2, 'copy') layer.add_node(BiasNode()) self.assertEqual(5, layer.total_nodes()) self.assertEqual(2, layer.total_nodes('input')) self.assertEqual(2, layer.total_nodes('copy')) self.assertEqual(0, layer.total_nodes('hidden'))