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]))