如何在python中加载文件中的数据,进行单元测试？

Question

我写了一个专门的HTML解析器,我想用我下载的几个示例网页进行单元测试.

在Java中,我使用了类资源,将数据加载到单元测试中,而不必依赖它们在文件系统的特定路径上.有没有办法在Python中执行此操作？

我找到了doctest.testfile()函数,但这似乎是doctests特有的.我想获得一个特定HTML文件的文件句柄,该文件与当前模块相关.

在此先感谢您的任何建议!

Answer 1

要从unittest中的文件加载数据,如果testdata与unittests在同一个目录上,则有一个解决方案:

TESTDATA_FILENAME = os.path.join(os.path.dirname(__file__), 'testdata.html')


class MyTest(unittest.TestCase)

   def setUp(self):
       self.testdata = open(TESTDATA_FILENAME).read()

   def test_something(self):
       ....

Answer 2

这是基于 Ferran 的回答，但它会在此期间关闭文件MyTest.tearDown()以避免“ResourceWarning: unclosed file”：

TESTDATA_FILENAME = os.path.join(os.path.dirname(__file__), 'testdata.html')


class MyTest(unittest.TestCase)

   def setUp(self):
       self.testfile = open(TESTDATA_FILENAME)
       self.testdata = self.testfile.read()

   def tearDown(self):
       self.testfile.close()

   def test_something(self):
       ....