在 PyTest 中创建一个临时目录

Question

我的 Python 项目导入 pytest 2.9.0 没有任何问题。

我想创建一个新的空目录，该目录仅在测试会话期间有效。我看到 pytest 提供临时目录支持：

You can use the tmpdir fixture which will provide a temporary directory unique to the test invocation, created in the base temporary directory.

tmpdir is a py.path.local object which offers os.path methods and more. Here is an example test usage:

pytest 的源代码显示 def tmpdir 是一个 global/module 函数：https://pytest.org/latest/_modules/_pytest/tmpdir.html

但是我的测试文件失败了：

import pytest

# ...

def test_foo():
    p = pytest.tmpdir()

有错误：

AttributeError: 'module' object has no attribute 'tmpdir'

执行 from pytest import tmpdir 失败：

ImportError: cannot import name tmpdir

Answer 1

您只需将 tmpdir 作为函数参数传递，因为它是 py.test 固定装置。

def test_foo(tmpdir):
    # do things with tmpdir

Answer 2

更新： 使用 tmp_path 而不是 tmpdir。 tmp_path 是一个 pathlib.Path/pathlib2.Path. tmpdir is a py.path (Actually LocalPath), which has offered syntax very similar to pathlib.Path. See pytest issue.

开发人员不再推荐使用 py.path。

语法类似，例如：

def test_something_else(tmp_path):
    #create a file "myfile" in "mydir" in temp directory
    f1 = tmp_path / "mydir/myfile"
    f1.parent.mkdir() #create a directory "mydir" in temp folder (which is the parent directory of "myfile"
    f1.touch() #create a file "myfile" in "mydir"


    #write to file as normal 
    f1.write_text("text to myfile")

    assert f1.read() == "text to myfile"

原文：我调查了一下，也发现了这种行为很奇怪，我总结了下面我学到的东西，以供那些不那么直观的人使用。

tmpdir 是 pytest 中的预定义夹具，类似于此处定义 setup 的方式：

import pytest

class TestSetup:
    def __init__(self):
        self.x = 4

@pytest.fixture()
def setup():
    return TestSetup()

def test_something(setup)
    assert setup.x == 4

因此 tmpdir 是在 pytest 中定义的固定名称，如果您将其作为参数名称传递给您的测试函数。

用法示例：

def test_something_else(tmpdir):
    #create a file "myfile" in "mydir" in temp folder
    f1 = tmpdir.mkdir("mydir").join("myfile")

    #create a file "myfile" in temp folder
    f2 = tmpdir.join("myfile")

    #write to file as normal 
    f1.write("text to myfile")

    assert f1.read() == "text to myfile"

这在您使用 pytest 运行时有效，例如在终端中运行ning py.test test_foo.py。以这种方式生成的文件具有读写权限，稍后可以在您的系统临时文件夹中查看（对我来说这是 /tmp/pytest-of-myfolder/pytest-1/test_create_file0）

在 PyTest 中创建一个临时目录

Creating a temporary directory in PyTest

python

pytest