(Additional) Better reference data

[constantinpape]

We are currently using this image as test data.
It might be a good idea to add some more complex data with maybe more dimensions and irregular shapes.

[jakirkham]

Maybe this would work. The height and width are different. It also contains an RGB channel, which may make it a bit more interesting. Plus it's pretty easy to check if it is orientated correctly.

[constantinpape]

I like the suggestion.
I will change it in the next few days.

[joshmoore]

An additional option would be to fairly small synethic arrays with some algorithm to decide whether or not each cell is correct:

import itertools
primes = [2, 3, 5, 7, 11]
shape = [5, 5, 5, 5, 5]

assert len(shape) <= len(primes)

for idx in itertools.product(*[range(s) for s in shape]):
    total = 1
    for prime, power in enumerate(idx):
        total *= (primes[prime]) ** (power+1)
    print(total)