def test_search_for_shawshank(self):
    "Using the search field, search for 'shawshank' and assert that shawshank was found."
    header_page = pagemodels.headerpage.HeaderPage(self.driver)
    header_page.search('shawshank')
    self.assertIn('The Shawshank Redemption', self.driver.page_source)
Using the search field, search for 'shawshank' and assert that shawshank was found.
def test_search_for_shawshank(self):
    
    header_page = pagemodels.headerpage.HeaderPage(self.driver)
    header_page.search('shawshank')
    self.assertIn('The Shawshank Redemption', self.driver.page_source)