Add tests for the "graph" command