Python 单元测试 while 循环

Question

我正在想办法怎么为一个输出基本字符串的while循环创建一个测试套件。我有一个很简单的倒计时器，只想测试一下它是否正常工作。我之前做过单元测试，但从来没有针对循环做过测试。下面是我想测试的代码。我只想得到一个'OK'的反馈。提前感谢大家的帮助！

import time

timer = 11

while timer > 0:
        timer = timer - 1
        print(timer)
        time.sleep(1)

print("Times up!")

Answer 1

我不太明白你想比较什么来判断测试是失败还是通过。不过，如果你是在找单元测试的代码片段，那我这儿有一个给你。

这是一个非常简单粗暴的检查方法。

import unittest
import time

expectedOutput = range(5)
class UnitTest(unittest.TestCase):

    def test_while(self):
        # Put your code here.
        timer = 5
        while timer > 0:
            timer = timer - 1 
            self.assertTrue(timer == expectedOutput[timer])

if __name__ == "__main__":    
    suite = unittest.TestLoader().loadTestsFromTestCase(UnitTest)
    unittest.TextTestRunner(verbosity=2).run(suite)