帶范圍功能的單元測試

Question

我會盡量保持清楚。 我已經練習單元測試已有一段時間了，現在我正試圖弄清楚如何測試以下代碼。

def main():

    for i in range(100):
        print("Argh!")

所以我只是想測試“ Argh！”的100次迭代！ 確保標准輸出通過測試。 我什至不知道如何開始單元測試。 關於這一點，雖然看起來很簡單。

預先謝謝大家。

Answer 1

假設您使用的是python3，則通過模擬即可輕松實現...

import unittest
import mock  # possibly "from unittest import mock" depending on version.
import main_module

class TestMain(unittest.TestCase):
    def test_main(self):
        with mock.patch.object(main_module, 'print') as mock_print:
            mock_module.main()
        expected_calls = [mock.call('Argh!') for _ in range(100)]
        mock_print.assert_has_calls(expected_calls)

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