检查页面a标签

<pre name="code" class="python"># coding=utf-8import unittestimport timefrom selenium import webdriverclass BaiduPageTest(unittest.TestCase):    def setUp(self):        self.driver = webdriver.Chrome()    def test_main(self):        driver = self.driver        driver.get("http://www.baidu.com")        time.sleep(2)         # 当前窗口句柄        currentHandle = driver.current_window_handle        links = driver.find_elements_by_tag_name("a")        time.sleep(1)        # driver.find_element_by_link_text("登录").click()        dr=driver.find_element_by_name("tj_login")        if(dr):            dr.click()        time.sleep(2)        # 关闭弹出框        # driver.find_element_by_id("TANGRAM__PSP_2__closeBtn").click()        #在弹出窗中输入数据并登陆        # driver.find_element_by_id("TANGRAM__PSP_8__userName").send_keys("[e-mail]")        # driver.find_element_by_id("TANGRAM__PSP_8__password").send_keys("baidu123")        # driver.find_element_by_id("TANGRAM__PSP_8__submit").click()        # # 得到所有窗口的句柄        # handles = driver.window_handles        time.sleep(5)        # self.assertIn("百度一下，你就知道",driver.title)    def tearDown(self):        self.driver.close()if __name__ == "__main__":    unittest.main()