使python py.test单元测试独立于py.test执行的位置运行?

时间:2015-12-02 16:18:58

标签: python unit-testing pytest

让我们说我的代码看起来像这样

import pytest
import json

@pytest.fixture
def test_item():
    test_item = json.load(open('./directory/sample_item_for_test.json','rb'))
    return test_item

def test_fun(test_document):
    assert  type(test_item.description[0]) == unicode

我想通过Py.Test运行此测试

如果我从它所在的目录运行Py.test,那很好。但是如果我从上面的目录运行它,由于无法找到' sample_item_for_test.json'而失败。有没有办法让这个测试正确运行,无论我在哪里执行Py.test?

2 个答案:

答案 0 :(得分:2)

魔术属性__file__是文件系统上python文件的路径。因此,您可以使用os中的一些魔法来获取当前目录...

import pytest
import json
import os

_HERE = os.path.dirname(__file__)
_TEST_JSON_FILENAME = os.path.join(_HERE, 'directory', 'sample_item_for_test.json')

@pytest.fixture
def test_item():
    with open(_TEST_JSON_FILENAME, 'rb') as file_input:
        return json.load(file_input)

答案 1 :(得分:1)

当我迁移到py.test时,我有一大堆遗留测试,习惯于在测试文件所在的目录中执行。我没有追踪每一个测试失败,而是在每个测试开始之前向我的conftest.py添加了一个pytest钩子到chdir到测试目录:

import os
import functools

def pytest_runtest_setup(item):
    """
    Execute each test in the directory where the test file lives.
    """
    starting_directory = os.getcwd()
    test_directory = os.path.dirname(str(item.fspath))
    os.chdir(test_directory)

    teardown = functools.partial(os.chdir, starting_directory)
    # There's probably a cleaner way than accessing a private member.
    item.session._setupstate.addfinalizer(teardown, item)