单元测试(Python),测试函数是否返回数组

时间:2013-12-09 11:51:14

标签: python unit-testing

我有一个python函数somefunc(),它必须返回一个像[1, 2, 3]这样的数组。现在我正在编写一个单元测试来检查这个函数返回的内容:

class TestingFunctionsTest(unittest.TestCase):
    def test_somefunc(self):
        #what to write here ?

我是python测试的新手,请帮助。

1 个答案:

答案 0 :(得分:1)

你可以写:

from mymodule import somefunc

class TestingFunctionsTest(unittest.TestCase):
    def test_somefunc(self):
        self.assertEqual(somefunc(), [1, 2, 3])

调用该函数并断言返回值必须等于[1, 2, 3]