Added test case for grid_shape parameter.