def test_valid_color(): num_nodes = 6 delta_mat = [[0,1,1,1,1,1],[1,0,1,1,1,1],[1,1,0,1,1,1],[1,1,1,0,1,1],[1,1,1,1,0,1],[1,1,1,1,1,0]] num_ants = 1 num_iterations = 10 colors1 = "RRRRRR" colors2 = "RRRBBB" colors3 = "BBBBBB" graph1 = AntGraph(num_nodes,delta_mat,colors1) colony1 = AntColony(graph1,num_ants,num_iterations) ant1 = Ant(1, 1, colony1) ant1.last_3_colors = ("R","R","R") print "False: " + str(ant1.valid_color(1)) ant1.last_3_colors = ("B","B","B") print "True: " + str(ant1.valid_color(1)) ant1.last_3_colors = ("R","R","B") print "True: " + str(ant1.valid_color(1)) graph2 = AntGraph(num_nodes,delta_mat,colors2) colony2 = AntColony(graph2,num_ants,num_iterations) ant2 = Ant(1, 1, colony2) ant2.last_3_colors = ("R","R","R") print "True: " + str(ant2.valid_color(3)) print "False: " + str(ant2.valid_color(1)) graph3 = AntGraph(num_nodes,delta_mat,colors3) colony3 = AntColony(graph3,num_ants,num_iterations) ant3 = Ant(1, 1, colony3) ant3.last_3_colors = ("R","R","R") print "True: " + str(ant3.valid_color(1)) ant3.last_3_colors = ("B","B","B") print "False: " + str(ant3.valid_color(1)) ant3.last_3_colors = ("B","R","B") print "True: " + str(ant3.valid_color(1))
def test_completable_path(): num_nodes = 8 delta_mat = [[0,1,1,1,1,1,1,1],[1,0,1,1,1,1,1,1],[1,1,0,1,1,1,1,1],[1,1,1,0,1,1,1,1],[1,1,1,1,0,1,1,1],[1,1,1,1,1,0,1,1],[1,1,1,1,1,1,0,1],[1,1,1,1,1,1,1,0]] num_ants = 1 num_iterations = 10 colors1 = "RRRRBBBB" graph1 = AntGraph(num_nodes,delta_mat,colors1) colony1 = AntColony(graph1,num_ants,num_iterations) ant1 = Ant(1, 1, colony1) print "testing reds_left initialization: " + str(ant1.reds_left) #Visited: 1,2,3 ant1.currNode = 3 ant1.last_3_colors = ("R","R","R") self.reds_left = 1