Add test_graph.py