Fixed pep8 error in test_graph