Add tests for search_model