检查函数中的值单元测试

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，尽管不能使用单元测试。您可以使用Python的内置assert关键字。TestCase.assertSomething这与unittest模块附带的功能不同，它可以在代码中的任何地方使用。例如，

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）