ENH: Better test coverage

Not ready for review/merge yet.
I made the example more clearly show the differences along dimensions.