def test_edges(self):
json_graph = {"label": "my graph", "graph": {"A": ["B"], "B": []}}
graph = Graph(input_graph=json.dumps(json_graph))
self.assertEqual(json_graph['label'], graph.get_label())
self.assertEqual(False, graph.is_directed())
self.assertEqual(json_graph['graph'], graph.get_graph())
self.assertEqual([Edge('A', 'B')], graph.edges())
def test_read_graph_from_file(self):
json_graph = {"label": "my graph", "graph": {"A": ["B"], "B": []}}
with open(path.join(self.test_dir, 'test.txt'), 'w') as out_file:
out_file.write(json.dumps(json_graph))
graph = Graph(input_file=str(path.join(self.test_dir, 'test.txt')))
self.assertEqual(json_graph["label"], graph.get_label())
self.assertEqual(False, graph.is_directed())
self.assertEqual(json_graph["graph"], graph.get_graph())
def save_to_json(graph: Graph, out_dir):
"""
:param graph: the graph to write to json
:param out_dir: the absolute path to the dir to write the file
:return:
"""
g_dict = {
"label": graph.get_label(),
"directed": graph.is_directed(),
"graph": graph.get_graph()
}
with open(path.join(out_dir, f"{graph.get_label()}.json"),
'w',
encoding="utf8") as out:
out.write(json.dumps(g_dict))
