TST: rework tests for explicit shapes