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

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

但是！您仍然可以检查Gold的值，但不能使用unittest。您可以使用内置assert关键字的Python。这与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关键字实际上是非常良好的编程风格。它不是单元测试的替代品，也不是完全相同的东西，而是它与单元测试一起工作以验证代码中的不变量。这意味着更快地捕获错误！

这里概述了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）

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

检查功能单元测试中的值

2 个答案: