In tests.test_grid improving tests