Pass grid_shape kwarg where appropriate

closes issue #235
Regression test for issue #235
Fix `grid_shape`s that were missed in PR #232's changes.
pep8 paralleltest_distributed_array_protocol.py
Merge pull request #257 from cowlicks/grid-shape-bug-gh-235

Don't ignore `grid_shape` kwarg