最近在學(xué)習(xí)unittest單元測試框架，之前調(diào)試好的測試腳本，發(fā)現(xiàn)放在這個框架中，無法正常switch_to.window，總是報錯無法定位到元素，個人感覺是因為窗口梅正常切換過去，新人求指教，謝謝，代碼如下：

main.py

import time
import unittest
import auto_test
suite = unittest.TestSuite()
suite.addTest(auto_test.Python_autotest('get_browser'))
suite.addTest(auto_test.Python_autotest('search'))
suite.addTest(auto_test.Python_autotest('user_login'))
suite.addTest(auto_test.Python_autotest('user_logout'))
if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(suite)

auto_test.py

from selenium import webdriver
import time
import unittest
class Python_autotest(unittest.TestCase):
    def setUp(self,driver = webdriver.Firefox()):
        print('Test is start')
        self.driver = driver
    def get_browser(self):
        self.driver.get("https://www.baidu.com/")
    def search(self):
        try:
            self.driver.find_element_by_xpath('//*[@id="kw"]').send_keys('segmentfault')
            self.driver.find_element_by_xpath('//*[@id="su"]').click()
        except Exception:
            print('No found everything, sorry!')
        else:
            time.sleep(1)
            self.driver.find_element_by_xpath("/html/body/div[1]/div[5]/div[1]/div[3]/div[1]/h3/a").click()
    def user_login(self):
        self.driver.switch_to.window(self.driver.window_handles[-1])
        time.sleep(3)
        self.driver.find_element_by_xpath('/html/body/div[4]/div/div/div[1]/div[2]/a[1]').click()#點擊頁面中的前端菜單欄
    def user_logout(self):
        self.driver.quit()
    def tearDown(self):
        print('Test is finish')
if __name__ == '__main__':
    unittest.main()