py+selenium IE 用driver.close()却把两个窗口都关了【已解决】

环境：py3  selenium  unittest

测试浏览器：IE10

目标：在单个文件中，有多个用例，执行完A用例，由于打开了新的窗口，必须关闭新的窗口，才不会影响下一条用例的执行。

问题：按例子1的代码，运行test001_a后，直接把两个窗口全部关闭，导致test002_b无法运行报错（以下只是例子，不一定会失败，实际运行系统时由于性能，网络等原因肯可能会失败）。

例子1：

 # coding:utf-8
 import unittest
 from selenium import webdriver

 class Testabc(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
         # 打开IE浏览器
         cls.driver = webdriver.Ie()
         # 最大化浏器
         cls.driver.maximize_window()

     @classmethod
     def tearDownClass(cls):
         cls.driver.quit()

     def test001_a(self):
         print(1)
         # 打开新的窗口
         self.driver.execute_script('window.open("https://www.sogou.com");')
         self.driver.switch_to.window(self.driver.window_handles[-1])  # 切换到当前最新打开的窗口
         self.driver.close()  # 关闭当前窗口
         self.driver.switch_to.window(self.driver.window_handles[-1])  #切换到当前最新打开的窗口

     def test002_b(self):
         print(2)

 if __name__ == "__main__":

     suite = unittest.TestLoader().loadTestsFromTestCase(Testabc)
     unittest.TextTestRunner(verbosity=2).run(suite)

若出现窗口全部被关闭，test002_b未执行，报错

则原因应该是，line22-24，由于代码运行速度较快，执行关闭后，窗口并未马上关闭，但已经执行了下一句，又切换到那个关闭的窗口，而运行第二条用例时，找不到第一个窗口的元素，因为没有切换到这个窗口！！

所以解决方案：

1，加入等待时间

2、如果只有两个窗口，直接关闭窗口后切换到第一个窗口

如图：

引申：

如果测试有经常打开窗口的，可以将这个放到teardown（）里，每次执行完毕后，关闭当前窗口，再切换第一个窗口。

问：那我有些用例只有一个窗口在跑怎么办？

答：做一个句柄判断。

其他情况就自己再造化了~~

 如有错误，请及时指出，谢谢！