ENH: Better test coverage

Not ready for review/merge yet.
I am pretty sure it's `(W, H, D)`