代码封装

641334778 · 发表于 2018-9-26 13:35:32

请问由于产品的每个功能模块都需要setUP中的代码，现在想把这部分代码分离出一个文件，但是分离后怎么才能使用dr属性呢
from selenium import webdriver
import time
import unittest
import HTMLTestRunner

class LoginCase(unittest.TestCase):
def setUp(self):
      self.dr = webdriver.Ie()
      self.dr.get("https://192.168.19.201")
      self.dr.maximize_window()
      self.dr.get("javascript:document.getElementById('overridelink').click();")
def tearDown(self):
      self.dr.quit()