Python测试中如何修改内置open函数的文件路径？

你猜的没错，这个TypeError确实是因为wb参数导致的——你的fake函数只定义了file_name一个参数，但实际调用open(output_file, 'wb')时，会传递两个 positional 参数（文件名和模式），如果原代码里还有其他可选参数（比如encoding、buffering），参数数量会更多，自然就会触发"参数不匹配"的错误。

下面是修正后的代码，同时完美实现"把文件写入到指定测试路径"的需求：

import pytest
import builtins  # 别忘了导入builtins模块，否则找不到原始的open方法

@pytest.fixture
def mock_output_file(mocker):
    orig_open = builtins.open
    def fake_open(*args, **kwargs):
        # 捕获所有传入open的参数，替换第一个参数（文件名）
        new_args = list(args)
        # 把原函数指定的文件名替换成测试用的文件
        new_args[0] = 'xml_test_file.xml'
        # 调用原始的open方法，传入修改后的参数
        return orig_open(*new_args, **kwargs)
    mocker.patch('builtins.open', new=fake_open)

class SomeClass():
    def run(self):
        output_file ="xml_initial_file.xml"
        print("Output file= ",output_file)
        with open(output_file,'wb') as f:
            # 这里可以写实际的文件内容，比如f.write(b'test content')
            pass

def test_change_name(mock_output_file):
    class_instance = SomeClass()
    class_instance.run()
    # 可选：添加断言验证测试文件是否被正确写入
    with open('xml_test_file.xml', 'rb') as f:
        content = f.read()
        # 比如断言内容符合预期：assert content == b'test content'

关键修改点说明： #

1. 用*args和**kwargs接收所有参数：这样不管原代码怎么调用open（传mode、encoding还是其他参数），都能完美兼容，不会再出现参数不匹配的错误。
2. 动态替换文件名：把args转成列表（因为元组不可变），修改第一个元素（也就是传入的文件名）为你想要的测试文件路径，再传给原始的open方法。
3. 建议用with语句管理文件：虽然这不是必须的，但能确保文件被正确关闭，避免资源泄漏。

如果想让测试更灵活（比如不同用例替换不同的测试文件），可以把fixture改成可参数化的形式：

@pytest.fixture
def mock_output_file(mocker):
    def _mock(target_file):
        orig_open = builtins.open
        def fake_open(*args, **kwargs):
            new_args = list(args)
            new_args[0] = target_file
            return orig_open(*new_args, **kwargs)
        mocker.patch('builtins.open', new=fake_open)
    return _mock

# 测试时指定不同的目标文件
def test_change_to_test_file(mock_output_file):
    mock_output_file('xml_test_file.xml')
    class_instance = SomeClass()
    class_instance.run()

def test_change_to_another_file(mock_output_file):
    mock_output_file('xml_another_test.xml')
    class_instance = SomeClass()
    class_instance.run()

这样就能轻松测试不同文件路径下的函数行为了。

内容的提问来源于stack exchange，提问作者viktor_dmitry