Improved the graph_suite function for tests.