愚弄导入的脚本,以便导入不存在的模块

时间:2015-10-15 20:11:45

标签: python python-import

我正在编写使用插件(在rules/plugins/目录中)的 foobaz python应用程序,并从项目 foobar rules/tests/}中测试这些插件STRONG>。我不控制这些(插件及其测试)。

由于测试中的import语句,我无法弄清楚如何运行这些语句。测试看起来像这样:

$ cat rules/tests/test_something.py
import unittest

from foobar.rules.plugins import something

class TestSomething(unittest.TestCase):
    def test_something_equals(self):
        self.assertEqual(1, something.equals(1))

插件本身可能如下所示:

$ cat rules/plugins/something.py
def equals(i):
  return i

我用来运行所有测试的脚本如下所示:

$ cat tests.py 
#!/usr/bin/env python

import sys
import unittest

sys.path.append('rules/tests/')
from test_something import TestSomething

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

当我想执行测试时,我(毫不奇怪)得到:

$ ./tests.py 
Traceback (most recent call last):
  File "./tests.py", line 8, in <module>
    from test_something import TestSomething
  File "rules/tests/test_something.py", line 6, in <module>
    from foobar.rules.plugins import something
ImportError: No module named foobar.rules.plugins

鉴于我无法编辑插件或测试,是否有某些方法可以让我运行?

如果我可以编辑测试,我会这样做:

- from foobar.rules.plugins import something
+ import sys
+ sys.path.append('rules/plugins/')
+ import something

它可以工作(测试)。

1 个答案:

答案 0 :(得分:1)

好的,在与同事讨论之后,有一个非常简单的方法来解决这个问题:

$ mkdir foobar
$ mv rules/ foobar/
$ touch foobar/__init__.py
$ touch foobar/rules/__init__.py
$ touch foobar/rules/plugins/__init__.py

这种方式测试将能够找到它需要的东西。我之前没有意识到这件事。