def test_adding_node(self):
""" Tests adding node operation. """
# Setup
genome = Genome()
genome.initialize(3, 1)
genome_connections = genome.get_connections()
# List of enabled (respectively, disabled) connections only.
init_true_connections = list(
filter(lambda a: a.is_enabled() == True, genome_connections))
init_false_connections = list(
filter(lambda a: a.is_enabled() == False, genome_connections))
init_connections_length = len(genome_connections)
# Run
genome.add_node()
# Assert
self.assertTrue(
len(
list(
filter(lambda x: x.is_enabled() == True,
genome.get_connections()))),
len(init_true_connections) + 1)
self.assertTrue(
len(
list(
filter(lambda x: x.is_enabled() == False,
genome.get_connections()))),
len(init_false_connections) + 1)
self.assertEqual(len(genome.get_connections()),
init_connections_length + 2)
def test_weight_mut():
a = Genome()
a.add_node(Node(NodeType.inp, 1))
a.add_node(Node(NodeType.inp, 2))
a.add_node(Node(NodeType.out, 3))
a.add_connection(Connection(1, 3, 0.5, True, 1))
a.add_connection(Connection(2, 3, 1.0, True, 2))
a.conn_innovation = 3
a.node_innovation = 4
a.render("render/weight_begin.gv")
a.mutate_weights()
a.render("render/weight_mutated.gv")
def test_crossover():
a = Genome()
for i in range(3):
a.add_node(Node(NodeType.inp, i + 1))
a.add_node(Node(NodeType.out, 4))
a.add_node(Node(NodeType.hidden, 5))
a.add_connection(Connection(1, 4, 1.0, True, 1))
a.add_connection(Connection(2, 4, 1.0, False, 2))
a.add_connection(Connection(3, 4, 1.0, True, 3))
a.add_connection(Connection(2, 5, 1.0, True, 4))
a.add_connection(Connection(5, 4, 1.0, True, 5))
a.add_connection(Connection(1, 5, 1.0, True, 8))
a.conn_innovation = 9
a.node_innovation = 6
b = Genome()
for i in range(3):
b.add_node(Node(NodeType.inp, i + 1))
b.add_node(Node(NodeType.out, 4))
b.add_node(Node(NodeType.hidden, 5))
b.add_node(Node(NodeType.hidden, 6))
b.add_connection(Connection(1, 4, 1.0, True, 1))
b.add_connection(Connection(2, 4, 1.0, False, 2))
b.add_connection(Connection(3, 4, 1.0, True, 3))
b.add_connection(Connection(2, 5, 1.0, True, 4))
b.add_connection(Connection(5, 4, 1.0, False, 5))
b.add_connection(Connection(5, 6, 1.0, True, 6))
b.add_connection(Connection(6, 4, 1.0, True, 7))
b.add_connection(Connection(3, 5, 1.0, True, 9))
b.add_connection(Connection(1, 6, 1.0, True, 10))
a.conn_innovation = 11
a.node_innovation = 7
c = Genome.crossover(b, a)
a.render("render/crossover_a.gv")
b.render("render/crossover_b.gv")
c.render("render/crossover_c.gv")
