Update graph_tests.go