If this restriction is really required, it should be mentioned the docstring.
Couldn't you use ndindex instead, and leave the door open for nD?
(That doesn't necessarily mean you have to write tests for nD, as we don't care about that right now.)