如何进行单元测试以检查 Python 中的输入不为空

Question

这是我第一次尝试用 Python 编写单元测试。我有一个像这样的简单函数：

def sum_num(a, b):
  return a+b

我想做单元测试，检查输入（a，b）不为空，输出不为空。

import unittest

class SumTest(unittest.TestCase):
    def test_sum_output_not_null(self):
        self.assertTrue(add_num(3,4))

    def test_sum_input_not_null(self):
        # How to check input (a and b) is not None ?
        self.assertIsNotNone(a)
...

suite = unittest.TestLoader().loadTestsFromTestCase(SumTest)
runner = unittest.TextTestRunner(verbosity=2)
runner.run(suite)

我在单元测试运行中遇到错误..

test_sum_input_not_null (__main__.SumTest) ... ERROR
test_sum_output_not_null (__main__.SumTest) ... ok

======================================================================
ERROR: test_sum_input_not_null (__main__.SumTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "<command-1933936>", line 7, in test_sum_input_not_null
    self.assertIsNotNone(a)
NameError: name 'a' is not defined

----------------------------------------------------------------------
Ran 2 tests in 0.000s

FAILED (errors=1)
Out[4]: <unittest.runner.TextTestResult run=2 errors=1 failures=0>

如何检查 a 和 b 不为空？也可能想检查 a 和 b 是否也是整数。我在某处读到有关 setup() 的信息。我需要这样做来测试函数的输入吗？

Answer 1

我想做单元测试，检查输入（a，b）不为空，输出不为空。

要么你还不明白测试的目的，要么你在问一些与 TDD 相关的问题。

你不测试引用，你测试当这种情况发生时你的函数是否处理好。

因此，您应该创建测试函数并在其中调用：

def test_when_a_is_null(self):
    self.assertIsNotNone(add_num(None, 5))

当 b 为 None 并且两者都为 None 时类似。

但是，这意味着你的函数应该处理条件：

def add_num(a, b):
    if a is not None and b is not None:
        return a + b
    elif a is not None:
        return a
    elif b is not None:
        return b
    return 0