在函数单元测试中检查值

Question

我想知道如何创建一个单元测试，来检查一个在函数内部的变量的值？我下面的代码只在变量是全局的情况下有效，我明白这个道理，但我并不想这样做。我只是想测试这个变量在主函数内部的值。我该怎么做呢？这是我的代码：

运行代码

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()

Answer 1

你现在的做法是不对的，也不应该这样做。不过，你可以使用一个闭包来返回一个函数，这样你就可以在之后调用它：

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()

Answer 2

据我所知，你无法访问函数内部的局部变量。实际上，这些变量在函数外部是不存在的。即使你能做到这一点，可能也不是你想要的效果。更好的做法是把函数当作独立的功能来对待，确保给定的输入能产生正确的输出和/或副作用。

不过！你仍然可以检查一下Gold的值，虽然不能用unittest来做。你可以使用Python内置的assert关键字。这和unittest模块里的TestCase.assertSomething函数不一样，assert可以在你代码的任何地方使用。例如，

def main():

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

def treasure_chest(amount):

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

有效地使用assert关键字其实是非常好的编程风格。它并不是单元测试的替代品，也不是完全相同的东西，而是和单元测试一起工作，用来验证你代码中的不变条件。这意味着可以更快地发现错误！

assert语句的语法在这里有说明，https://docs.python.org/3.4/reference/simple_stmts.html。简单来说，你给它一个布尔表达式，它会检查这个值是否为True。如果不是，它会抛出一个AssertionError。

通常，你只想用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）