Add a few more tests to TestGraph