Added docstrings to test_grid.py