pep8 test_search.py, modify import