Question

我正在编写读取文件的代码，并根据该文件的内容创建字典。代码非常简单，但我想测试边缘情况。

这是我的尝试：

from tempfile import NamedTemporaryFile
from nose.tools import *

def read_file(filename):
    with open(filename) as f:
        my_dict = { dict(line.strip().split(',')) for line in f }
    return my_dict

def test_read_file():
    file_contents = b"""Hello,World"""
    with NamedTemporaryFile() as fp:
        fp.write(file_contents)
        my_dict = read_file(fp.name)
    print(my_dict)
    assert my_dict == { "Hello" : "World" }

不幸的是，这个断言失败了，因为my_dict是一个空字典。

我的理解是，一旦NamedTemporaryFile被关闭，它就会被销毁，所以我不希望它在read_file之后<{>直接 my_dict {{} 1}}已填充。 fp被打开两次：一次写入，一次读取 - 这是麻烦制造者吗？

这是测试读取文件的函数的正确方法吗？如果是这样，为什么我的断言失败了？如果没有，那么编写此测试的更好的机制是什么？

Answer 1

您需要刷新写入以确保在读取之前写入数据。

def test_read_file():
    file_contents = b"""Hello,World"""
    with NamedTemporaryFile() as fp:
        fp.write(file_contents)
        fp.flush()
        my_dict = read_file(fp.name)
    print(my_dict)
    assert my_dict == { "Hello" : "World" }

Answer 2

您可以使用内存中类文件对象，而不是使用真实文件。这需要模拟open，或者更改您的API以获取类似文件的对象而不是文件名。

import unittest.mock
import io

def test_read_file():
    file_contents = io.BytesIO(b"""Hello,World""")
    m = unittest.mock.mock_open(read_data=file_contents)
    with unittest.mock.patch('__main__.open', m):
        my_dict = read_file("fake.txt")
    print(my_dict)
    assert my_dict == { "Hello" : "World" }

为什么我的断言在读取NamedTemporaryFile时失败了？

2 个答案: