python - 检查函数单元测试中的值

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。您可以使用 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 关键字实际上是真正好的编程风格。它不是单元测试的替代品，也不是做完全相同的事情，而是与单元测试一起工作以验证代码中的不变量。这意味着更快地捕获错误!

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

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

例如，

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)