Fix axis ij to both types of plots.
Make edge_test into a test with assert.