0

我试图弄清楚如何为我用 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 人——除此之外，任何其他的都会导致数字四舍五入。

谁能看到我如何让测试用例工作以及我编写测试代码的地方出错了？

4

1 回答 1

2

在您的单元测试文件中，您必须import num_buses.

这解决了你的直接问题，但如果你已经定义num_buses了，a1那么你必须这样做，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 测试- 只需按下恢复，然后test_untitled.py按下 TEST 按钮。

于 2013-04-06T07:35:12.247 回答