如何重用相同的函數，但在同一測試腳本中從python中的不同位置調用它

Question

我有一個測試用例，其中我必須從設備上的不同屏幕“結束進程”，並且具有模擬不同屏幕的功能。 EndProcess（）之后，設備返回到screen1（）。 有什么樂觀的方法可以在Python中做到這一點嗎？ 我可以在這里使用發電機嗎？

目前，我的代碼是：

while 1:
    screen1()
    EndProcess()
    screen1()
    screen2()
    EndProcess()
    screen1()
    screen2()
    screen3()
    EndProcess()

Answer 1

當屏幕數量變大時，您將重復很多次。 相反，您可以將屏幕放在列表中，並使用for循環調用它們：

screens = [screen1, screen2, screen3]

while True:
   for x in range(len(screens)):
      for i in range(x+1):
         screens[i]()
      EndProcess()

在Python 2中使用xrange代替range 。