使用Python中的外部数据文件进行单元测试

Question

我是python的新手，正在尝试运行注入了来自外部文件的数据的单元测试：

class TestIsEquivalent(unittest.TestCase):

    filename = os.getcwd() + '\\data\\test-data.txt'

    with open(filename) as f:
        testdata = f.readlines()

    for value in testdata:
        print(value)

        def test_0(self):
            result = isEquivalent(value, value, 0)
            self.assertEqual(result, True)

我在测试函数中调用isEquivalent（）时遇到以下错误：

NameError: name 'value' is not defined 

这似乎暗示了一个范围界定问题，但是我无法弄清楚如何将for循环中的值传递给测试方法。 任何建议将不胜感激。

Answer 1

您完整的测试用例应包装在一个函数中。 我不能完全确定您要在for循环中使用函数定义做什么，但是将def test_0(self)移到类的顶部可以解决问题。

class TestIsEquivalent(unittest.TestCase):
    def test_0(self):

        filename = os.getcwd() + '\\data\\test-data.txt'

        with open(filename) as f:
            testdata = f.readlines()

        for value in testdata:
            print(value)

            result = isEquivalent(value, value, 0)
            self.assertEqual(result, True)

但是，如果您要创建多个测试用例，使您都拥有testdata内的所有值，并且不想多次读取文件，则可以使用setUpClass函数，该函数在类内所有单元测试开始时执行一次。 结果看起来像这样。

class TestIsEquivalent(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        filename = os.getcwd() + '\\data\\test-data.txt'

        with open(filename) as f:
            cls.testdata = f.readlines()

    def test_0(self):
        for value in self.testdata:
            print(value)

            result = isEquivalent(value, value, 0)
            self.assertEqual(result, True)