def test_load_save(self):
algo = GraphAlgo()
self.assertFalse(algo.load_from_json("../data/sadasasdas.json"))
self.assertTrue(algo.load_from_json("../data/T0.json"))
self.assertTrue(algo.save_to_json("../data/Test.json"))
graph = algo.get_graph()
self.assertEqual(graph.v_size(), 4)
self.assertEqual(graph.e_size(), 5)
def my_graph(filename):
algo = GraphAlgo()
algo.load_from_json(filename)
start = timeit.default_timer()
algo.shortest_path(1, 5)
stop = timeit.default_timer()
print("my shoetest path ", (stop - start))
start = timeit.default_timer()
algo.connected_components()
stop = timeit.default_timer()
print("my connected components ", (stop - start))
start = timeit.default_timer()
algo.connected_component(0)
stop = timeit.default_timer()
print("my connected component ", (stop - start))
def test_plot(self):
algo = GraphAlgo()
self.assertTrue(algo.load_from_json("../data/A5"))
algo.plot_graph()