检查函数单元测试中的值答案

【问题标题】：Checking a value within a function Unit Testing检查函数单元测试中的值
【发布时间】：2014-08-20 00:31:04
【问题描述】：

我正在尝试找出如何创建单元测试来检查函数内变量的值？我有下面的代码，只有在我得到的变量是全局变量时才有效，但这不是我想要做的。我只想在主函数中测试变量的值。我该怎么做呢？这是我的代码：

运行代码

def main():
    #Declare Variable Here
    Gold = 4000

def treasure_chest(amount):
    print("I have ",amount," Gold!")

这是我的测试套件

from Functions import main
import unittest

class MyTestCase(unittest.TestCase):
    def test_valueContains(self):
        value =  Gold
        self.assertEquals(4000, value)

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

【问题讨论】：

这有点反模式。测试应将代码视为黑盒。例如“如果给定适当的输入，它是否提供正确的输出”？原因之一是代码的实现方式可能会发生变化。只要它完成与以前相同的工作，单元测试就不应该关心它的如何。一个人为的例子是将宝藏分成可能具有不同价值的黄金和宝石。 treasure_chest 函数可以正确计算该值，但您的单元测试可能仍会失败。

标签： python function unit-testing assertions

【解决方案1】：

据我所知，您无法访问函数本地的任何变量。事实上，它们实际上并不存在于该范围之外。即使你能做到这一点，它也可能不是你想要做的。最好将函数视为函数，并针对产生正确输出和/或副作用的给定输入验证这一点。

但是！您仍然可以检查 Gold 的值，尽管不能使用 unittest。您可以使用 Python 内置的 assert 关键字。这与 unittest 模块附带的TestCase.assertSomething 函数不同，它可以在代码中的任何地方使用。例如，

def main():

  #Declare Variable Here
  Gold = 4000
  assert Gold == 4000

def treasure_chest(amount):

    assert Gold <= somethingElse
    print("I have ",amount," Gold!")

有效地使用assert 关键字实际上是真正良好的编程风格。它不是单元测试的替代品，也不是完全相同的事情，而是与单元测试一起工作以验证代码中的不变量。这意味着更快地捕获错误！

断言语句的语法在此处概述，https://docs.python.org/3.4/reference/simple_stmts.html。用最简单的形式，你给它一个布尔表达式，它会检查值是True。如果不是，它将引发AssertionError。

通常您只想使用assert 来断言必须正确的值。一个常见的习惯用法是将它用于 assert 私有方法/函数的参数，因为我们假设这些方法的输入已经过清理并且应该是有效的。

例如，

class Foo():

    __init__(self, val):
       if val < 0:
           raise InvalidValueOrSomething

       __val = val

   __modify_val(self):
       assert self.__val < 0
       # Do some stuff...

请注意，默认情况下，断言始终运行，即使在不测试时也是如此。

可以在运行时使用解释器的-O 标志禁用断言，但不建议这样做 (https://wiki.python.org/moin/UsingAssertionsEffectively)

【讨论】：

太棒了！和很棒的细节。谢谢。我有点以为我做不到。只是想绝对确定。但是您使用内置 assert 方法的示例很棒。谢谢
@Salce__ 很高兴为您提供帮助！仅供参考assert 是在语言中创建特殊断言语句的关键字，而不是真正的方法（它是附加到对象的函数。）=）
@Salce__ 我也会在下面查看 Aaron 的答案。返回闭包非常整洁。一般来说，我可能不会测试一个变量，但了解闭包的工作原理会非常有帮助和乐趣！

【解决方案2】：

你不能也不应该按照你的方式来做，但是你可以用一个闭包来做，返回你可以稍后调用的函数：

def closure(Gold):
    Gold = Gold
    def treasure_chest():
        print("I have", Gold,"Gold!")

    return treasure_chest

测试套件

from Functions import closure
import unittest

class MyTestCase(unittest.TestCase):
    def setUp(self):
        self.amount = 4000
        self.treasure_chest = closure(self.amount)

    def test_valueContains(self):
        self.assertEquals(self.amount, self.treasure_chest.__closure__[0].cell_contents)

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

【讨论】：