pythonunittest：目录和文件路径

Question

对于我正在开发的公开keras模型的库，我有以下文件结构：

relevancy (repo)
    relevancy (package repo)
        data
            model.h5
            tokenizer.pickle
        test
            __init__.py
            test_model.py
        model.py
        __init__.py
    __init__.py
    setup.py

该库基本上加载预先训练的tokenizer.pickle和{}，并对输入数据进行预测。在

在model.py中，我有一个包含以下代码的函数，用于加载标记器和模型：

^{pr2}$

在test_model.py中，我在测试中调用这个函数。在

如果我在/relevancy (repo)下调用python setup.py test，我将得到一个错误消息，说找不到{}。显然，是亲戚造成的。在

我应该如何设置我的目录或路径，以便始终正确加载标记器和模型？在

Answer 1

如果需要访问存储在包中的数据文件，请考虑使用pkg_resources module。在

然后在model.py中，您可以执行以下操作：

filename = pkg_resources.resource_filename(__name__, 'data/tokenizer.pickle')
with open(filename, 'rb') as f:
    ...