def pruebas_grafo_mst(): # Ejemplo del capitulo 23 de Cormen grafo = Grafo(True) print_exception(grafo.componentes_conexas, TypeError, "Pruebas Grafo, mst de un grafo dirigido lanza TypeError") grafo = Grafo() print_test("Pruebas Grafo, MST de un grafo vacio es None", not grafo.mst()) grafo.add("A") grafo.add("B") grafo.add("C") grafo.add("D") grafo.add("E") grafo.add("F") grafo.add("G") grafo.add("H") grafo.add("I") grafo.agregar_arista("A", "B", 4) grafo.agregar_arista("A", "H", 8) grafo.agregar_arista("B", "C", 8) grafo.agregar_arista("B", "H", 8) grafo.agregar_arista("C", "D", 7) grafo.agregar_arista("C", "F", 4) grafo.agregar_arista("C", "I", 2) grafo.agregar_arista("D", "E", 9) grafo.agregar_arista("D", "F", 14) grafo.agregar_arista("E", "F", 10) grafo.agregar_arista("F", "G", 2) grafo.agregar_arista("G", "I", 6) grafo.agregar_arista("G", "H", 1) grafo.agregar_arista("H", "I", 7) n_grafo = grafo.mst() suma = [0, n_grafo] n_grafo.dfs(visitar_y_sumar, suma) print_test("Pruebas Grafo, la suma del mst conexo es correcta", suma[0] == 37) grafo.borrar_arista("B", "H") grafo.borrar_arista("B", "C") grafo.borrar_arista("H", "G") grafo.borrar_arista("H", "I") # el grafo ya no es conexo n_grafo = grafo.mst() # se deberian formar dos arboles suma = [0, n_grafo] n_grafo.dfs(visitar_y_sumar, suma, "A") # primer arbol print_test( "Pruebas Grafo, la suma del primer arbol del mst no conexo es correcta", suma[0] == 12) suma = [0, n_grafo] n_grafo.dfs(visitar_y_sumar, suma, "I") # segundo arbol print_test( "Pruebas Grafo, la suma del segundo arbol del mst no conexo es correcta", suma[0] == 24)
def pruebas_grafo_camino_minimo(): grafo = Grafo() print_exception( grafo.sssp, KeyError, "Pruebas Grafo, camino minimo con una clave que no existe en lanza KeyError", "A") grafo.add("A") grafo.add("B") grafo.add("C") grafo.add("D") grafo.add("E") grafo.add("F") grafo.add("G") grafo.add("H") grafo.add("I") grafo.agregar_arista("A", "B", 4) grafo.agregar_arista("A", "H", 8) grafo.agregar_arista("B", "C", 8) grafo.agregar_arista("B", "H", 8) grafo.agregar_arista("C", "D", 7) grafo.agregar_arista("C", "F", 4) grafo.agregar_arista("C", "I", 2) grafo.agregar_arista("D", "E", 9) grafo.agregar_arista("D", "F", 14) grafo.agregar_arista("E", "F", 10) grafo.agregar_arista("F", "G", 2) grafo.agregar_arista("G", "I", 6) grafo.agregar_arista("G", "H", 1) grafo.agregar_arista("H", "I", 7) print_test("Pruebas Grafo, camino minimo A-A es correcto", grafo.camino_minimo("A", "A") == ["A"]) print_test("Pruebas Grafo, camino minimo A-I es correcto", grafo.camino_minimo("A", "I") == ["A", "B", "C", "I"]) print_test("Pruebas Grafo, camino minimo A-E es correcto", grafo.camino_minimo("A", "E") == ["A", "H", "G", "F", "E"]) print_test("Pruebas Grafo, camino minimo I-A es correcto", grafo.camino_minimo("I", "A") == ["I", "C", "B", "A"]) print_test("Pruebas Grafo, camino minimo C-F es correcto", grafo.camino_minimo("C", "F") == ["C", "F"]) grafo.borrar_arista("B", "H") grafo.borrar_arista("B", "C") grafo.borrar_arista("H", "G") grafo.borrar_arista("H", "I") # el grafo ya no es conexo print_test("Pruebas Grafo, camino minimo A-C es None", not grafo.camino_minimo("A", "C")) print_test("Pruebas Grafo, camino minimo B-D es None", not grafo.camino_minimo("B", "D")) print_test("Pruebas Grafo, camino minimo F-H es None", not grafo.camino_minimo("F", "H"))
def pruebas_grafo_sssp(): grafo = Grafo() print_exception( grafo.sssp, KeyError, "Pruebas Grafo, sssp con una clave que no existe en lanza KeyError", "A") grafo.add("A") print_test("Pruebas Grafo, sssp en un grafo unitario es el mismo grafo", grafo.crear_grafo_con_padre(grafo.sssp("A")[0], True) == grafo) grafo.add("B") grafo.add("C") grafo.add("D") grafo.add("E") grafo.add("F") grafo.add("G") grafo.add("H") grafo.add("I") grafo.agregar_arista("A", "B", 4) grafo.agregar_arista("A", "H", 8) grafo.agregar_arista("B", "C", 8) grafo.agregar_arista("B", "H", 8) grafo.agregar_arista("C", "D", 7) grafo.agregar_arista("C", "F", 4) grafo.agregar_arista("C", "I", 2) grafo.agregar_arista("D", "E", 9) grafo.agregar_arista("D", "F", 14) grafo.agregar_arista("E", "F", 10) grafo.agregar_arista("F", "G", 2) grafo.agregar_arista("G", "I", 6) grafo.agregar_arista("G", "H", 1) grafo.agregar_arista("H", "I", 7) n_grafo = grafo.crear_grafo_con_padre(grafo.sssp("A")[0], True) suma = [0, n_grafo] n_grafo.dfs(visitar_y_sumar, suma, "A") print_test( "Pruebas Grafo, la suma de las distancias minimas desde 'A' a todos los vertices es correcta", suma[0] == 42) grafo.borrar_arista("B", "H") grafo.borrar_arista("B", "C") grafo.borrar_arista("H", "G") grafo.borrar_arista("H", "I") # el grafo ya no es conexo n_grafo = grafo.crear_grafo_con_padre(grafo.sssp("A")[0], True) suma = [0, n_grafo] n_grafo.dfs(visitar_y_sumar, suma, "A") print_test( "Pruebas Grafo, la suma de las distancias minimas desde 'A' a todos los vertices es correcta", suma[0] == 12)
def pruebas_grafo_varios_elementos(): grafo = Grafo() vars = ["iron", "maiden", "black", "sabbath"] for value in vars: grafo.add(value) print_test("Pruebas Grafo, la cantidad de elementos es cuatro", len(grafo) == 4) for value in vars: ok = value in grafo if not ok: break ok2 = value in grafo.keys() if not ok2: break print_test( "Pruebas Grafo, todos los elementos estan contenidos en el grafo", ok) print_test( "Pruebas Grafo, todos los elementos estan contenidos en el grafo las claves del grafo", ok2) grafo.agregar_arista(vars[0], vars[1]) # grafo.agregar_arista(vars[0], vars[2]) # |0|------|1| grafo.agregar_arista(vars[1], vars[2]) # | -. | grafo.agregar_arista(vars[1], vars[3]) # | -. | grafo.agregar_arista(vars[2], vars[3]) # |2|------|3| print_test("Pruebas Grafo, la arista existe", grafo.obtener_peso_arista(vars[0], vars[1])) print_test("Pruebas Grafo, la arista opuesta existe", grafo.obtener_peso_arista(vars[1], vars[0])) print_test("Pruebas Grafo, la arista existe", grafo.obtener_peso_arista(vars[0], vars[2])) print_test("Pruebas Grafo, la arista opuesta existe", grafo.obtener_peso_arista(vars[2], vars[1])) print_test("Pruebas Grafo, la arista existe", grafo.obtener_peso_arista(vars[1], vars[2])) print_test("Pruebas Grafo, la arista opuesta existe", grafo.obtener_peso_arista(vars[2], vars[1])) print_test("Pruebas Grafo, la arista existe", grafo.obtener_peso_arista(vars[1], vars[3])) print_test("Pruebas Grafo, la arista opuesta existe", grafo.obtener_peso_arista(vars[3], vars[1])) print_test("Pruebas Grafo, la arista existe", grafo.obtener_peso_arista(vars[2], vars[3])) print_test("Pruebas Grafo, la arista opuesta existe", grafo.obtener_peso_arista(vars[3], vars[2])) print_test("Pruebas Grafo, adyacentes ok", all(x in [vars[1], vars[2]] for x in grafo.adyacentes(vars[0]))) print_test( "Pruebas Grafo, adyacentes ok", all(x in [vars[0], vars[2], vars[3]] for x in grafo.adyacentes(vars[1]))) print_test( "Pruebas Grafo, adyacentes ok", all(x in [vars[0], vars[1], vars[3]] for x in grafo.adyacentes(vars[2]))) print_test("Pruebas Grafo, adyacentes ok", all(x in [vars[1], vars[2]] for x in grafo.adyacentes(vars[3]))) for vertice1 in grafo: # borro todas las aristas for vertice2 in grafo: try: grafo.borrar_arista(vertice1, vertice2) except ValueError: pass print_test("Pruebas Grafo, la cantidad de elementos es cuatro", len(grafo) == 4) grafo.agregar_arista(vars[0], vars[1]) # grafo.agregar_arista(vars[0], vars[2]) # |0|------|1| grafo.agregar_arista(vars[1], vars[2]) # | -. | grafo.agregar_arista(vars[1], vars[3]) # | -. | grafo.agregar_arista(vars[2], vars[3]) # |2|------|3| del grafo[vars[0]] print_test("Pruebas Grafo, el nodo borrado no pertenece al grafo", not vars[0] in grafo) print_exception( grafo.obtener_peso_arista, KeyError, "Pruebas Grafo, obtener peso arista borrada lanza KeyError", vars[0], vars[1]) print_exception( grafo.obtener_peso_arista, KeyError, "Pruebas Grafo, obtener peso arista reciproca borrada lanza KeyError", vars[1], vars[0]) print_test("Pruebas Grafo, la cantidad de elementos es tres", len(grafo) == 3) print_test( "Pruebas Grafo, la cantidad de elementos de grafo.keys() es tres", len(grafo.keys()) == 3) print_test("Pruebas Grafo, adyacentes ok", vars[0] not in grafo.adyacentes(vars[1])) print_test("Pruebas Grafo, adyacentes ok", vars[0] not in grafo.adyacentes(vars[2])) print_test("Pruebas Grafo, adyacentes ok", vars[0] not in grafo.adyacentes(vars[3]))