Update search tests for changes to search.py