Add a check on the grid shape

[rieder]

in Python 3.10, the 'else' check won't work if the shapes are unequal, so add this as an additional check.