Python - 如何从类调用外部函数然后返回到类

Question

好吧，澄清一下，我正在尝试使用 selenium 在网站上进行数据驱动测试: 主要类代码为:

from  a_folder.abc_file import userJourney
from  a_folder.a1_file import contact

@ddt
class testScenario(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome(path)

    def test_main_page(arg1, arg2):
         if arg1 == 'yes':
             return userJourney(arg1, arg2)

         if arg 2 == 'no':
             return contact(arg1, arg2)


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

参数从 csv 文件传入

userJourney() 和 contact() 函数是从外部文件导入的

我遇到的问题是 arg1 和 arg2 可以同时为 true，因此我希望执行两个函数(userJouney 和 contact)，但测试只执行一个函数，然后直接进入 TeaDownClass 函数。关于我缺少什么有什么想法吗？

abc_file.py 类似于

def userJourney(arg1, arg2):
   # Find element and click

提前致谢

Answer 1

只有一个方法会被执行，因为在这两种情况下都使用了“return”。因此，如果 arg1 == 'yes'，则将执行 userJourney 函数，之后 test_main_page 将退出并将控制权返回给调用者，因为您在调用 userJourney 之前使用了 'return' 关键字。下一个案例将不会执行，因为此时方法已完成。

您可能应该从这两种情况中删除“return”关键字。我不确定你想在这里测试什么，但那是另一回事。