def test_find_element_return_Element(self): element = Element("search button", "xpath") page = Page("google", "http://www.google.com") page.elements["search button"] = element assert page.find_element("search button") == element
def test_find_element_return_None(self): page = Page("google", "http://www.google.com") assert page.find_element("search button") == None