Question

我有一个标准的python proj 包::

proj/
├── proj
│   ├── data
│   │   └── data.csv
│   ├── __init__.py
│   └── lib.py
├── MANIFEST.in
├── setup.py
└── tests
    ├── __init__.py
    └── lib_test.py

data/data.csv与setup.py ::

中的 MANIFEST.in 和 include_package_data 键一起安装

~/proj$ more setup.py
 from setuptools import setup
 setup(name='proj',
       version=1.0,
       packages=['proj'],
       test_suite = 'tests',
       include_package_data=True,
       zip_safe=False)

~/proj$ more MANIFEST.in
include proj/data/*.csv

所以在python setup.py install data.csv位于site-packages ::

之后的地方之后

$ tree ~/venv/lib/python2.7/site-packages/proj-1.0-py2.7.egg
~/venv/lib/python2.7/site-packages/proj-1.0-py2.7.egg
├── EGG-INFO
│   └── ...
└── proj
    ├── data
    │   └── data.csv
    ├── __init__.py
    └── ...

在setup.py中，声明了 test_suite ，所以当我首先从proj调用test时文件夹，这项工作很好::

~/proj$ python setup.py test
running test
...
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK

我想在我的unitest中使用来自：〜/ venv / lib / python2.7 / site-packages / proj-1.0-py2.7.egg / proj / data / data的data.csv文件。 CSV。简而言之，可以获得〜/ venv / lib / python2.7 / site-packages / proj-1.0-py2.7.egg / proj / data / 文件夹。

使用此代码::

$ more tests/lib_test.py
import unittest, os, proj
from proj.lib import get_data
class TestUtils(unittest.TestCase):
    def test_get_data(self):
        datapath = os.path.dirname(os.path.abspath(proj.__file__)) + '/data'
        data = '{}/data.csv'.format(datapath)
        print '\n my data:', data

if __name__ == '__main__':
    unittest.main()

使用__file__我只获得〜/ sandbox / proj / proj / data / data.csv ::

$ python setup.py test
running test
...
test_get_data (tests.lib_test.TestUtils) ...
 my data: ~/sandbox/proj/proj/data/data.csv
ok
----------------------------------------------------------------------
Ran 1 test in 0.000s

Answer 1

实际上

python setup.py test

不会从python setup install放置它的地方加载python包（〜/ venv / lib / python2.7 / site-packages / proj-xxx），但是从python setup.py test运行的地方开始

这就是为什么：

import proj, os
os.path.abspath(proj.__file__)

我们无法捕获site-package文件夹，只能捕获工作副本文件。

从python unittest

1 个答案: