Added in tests for with-graph.