add a test for grid_search