将selenium python文件分成较小的块

时间:2018-02-20 23:41:21

标签: python selenium

我有几个用于测试我的页面的python文件,现在这些文件变得很多,并且它们都共享

def setUp(self):
    self.driver = webdriver.Chrome(chrome_options=options)

这个

def tearDown(self):
    self.driver.close()

if __name__ == "__main__":
    unittest.main()

是否可以在我的其他运行中导入这些?如果是这样,怎么样?

我试过在一个名为start.py的文件中然后使用

from start import setUp

但它没有工作

1 个答案:

答案 0 :(得分:1)

试试这个,创建一个基础测试类,并让所有测试类继承它。话虽如此,我不知道webdriver是否会很好地发挥(未经测试)

import unittest

class BaseTest(unittest.TestCase):

    def setUp(self):
        print('--- Setup ---')

    def tearDown(self):
        print('--- Tear Down ---')


class Test1(BaseTest):

    def test_equal(self):
        self.assertEqual(1, 1)


class Test2(BaseTest):

    def test_equal(self):
        self.assertEqual(2, 2)


if __name__ == '__main__':
    unittest.main()

<强>输出:

--- Setup ---
.--- Tear Down ---
--- Setup ---
--- Tear Down ---
.
----------------------------------------------------------------------
Ran 2 tests in 0.001s