如何使用unittest正確編寫測試？

Question

我試圖弄清楚如何為我用Python編寫的函數編寫單元測試 - 這里是下面的代碼：

def num_buses(n):
    import math
    """ (int) -> int

    Precondition: n >= 0

    Return the minimum number of buses required to transport n people.
    Each bus can hold 50 people.

    >>> num_buses(75)
    2
    """
    bus = int()
    if(n>=0):
        bus = int(math.ceil(n/50.0))
    return bus

我正在嘗試編寫測試代碼，但是他們給了我失敗的結果 - 這是我開始的代碼：

import a1
import unittest


class TestNumBuses(unittest.TestCase):
    """ Test class for function a1.num_buses. """
    def test_numbuses_1(self):
        actual = num_buses(75)
        expected = 2
        self.assertEqual(actual, expected)

    # Add your test methods for a1.num_buses here.


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

當我按F5運行模塊時 - 這就是我得到的 -

E
======================================================================
ERROR: test_numbuses_1 (__main__.TestNumBuses)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "C:\1-blog-cacher\TestNumBuses.py", line 8, in test_numbuses_1
    actual = num_buses(75)
NameError: global name 'num_buses' is not defined

----------------------------------------------------------------------
Ran 1 test in 0.050s

FAILED (errors=1)

所以我的測試失敗 - 雖然它應該通過，因為乘客的數量是75，每輛公共汽車最多可以容納50人 - 任何超過這個將導致數字的四舍五入。

任何人都可以看到我如何讓測試用例工作以及我在哪里編寫測試代碼出錯？

Answer 1

在您的unittest文件中，您必須import num_buses 。

這可以解決您的問題，但是如果您在a1定義了num_buses ，那么您必須執行a1.num_buses否則Python會認為num_buses是一個全局函數。

import a1
import unittest
import num_buses

class TestNumBuses(unittest.TestCase):
    """ Test class for function a1.num_buses. """
    def test_numbuses_1(self):
        actual = num_buses(75) #a1.num_buses(75) <-
        expected = 2
        self.assertEqual(actual, expected)

    # Add your test methods for a1.num_buses here.


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

看看這個： Cyber​​-Dojo測試 - 只需按下resume，然后在test_untitled.py按下TEST按鈕。