Add tests for normalise_grid_size