使用pytest测试src目录脚本时遇ModuleNotFoundError问题求助
搞定pytest的ModuleNotFoundError问题
嘿,这个问题我太熟了!本质就是你在test目录下跑pytest时,Python解释器的模块搜索路径里没包含src目录,所以找不到script2这些模块。下面给你几个实用的解决思路,从临时快速方案到长期规范方案都有:
方法一:临时把src目录加入Python路径
如果只是想快速跑通测试,直接在test_fun.py最顶部加几行代码,手动把src的路径塞到Python的搜索路径里:
import sys from pathlib import Path # 找到src目录的绝对路径——从test_fun.py往上跳两级到项目根目录,再定位到src src_dir = Path(__file__).parent.parent / "src" sys.path.append(str(src_dir)) # 现在就能正常导入啦 from script2 import fun2
这个方法简单粗暴,不用改项目结构,但缺点是每个测试文件都得加这段代码。
方法二:把src做成可安装的包(推荐!)
如果你的项目打算长期维护,这个是最规范的做法:
- 在项目根目录(就是包含
src和test的那个文件夹)创建一个pyproject.toml文件,内容如下:
[build-system] requires = ["setuptools>=61.0"] build-backend = "setuptools.build_meta" [project] name = "你的项目名" version = "0.1.0" packages = ["src"]
- 然后在项目根目录跑这个命令,把你的包安装成可编辑模式:
pip install -e .
之后你在测试文件里就能用绝对导入了,比如:
from src.script2 import fun2
甚至如果src/__init__.py里导出了模块,还能这么写:
from src import script2
这个方法是Python项目的标准操作,不仅解决测试问题,以后要打包分发也方便。
方法三:换个目录跑pytest
别在test目录下跑,回到项目根目录再运行pytest:
cd /你的项目根目录路径 pytest test/test_fun.py
同时记得给test目录也加个__init__.py(把它变成Python包),然后测试文件里用相对导入:
from ..src.script2 import fun2
不过相对导入有时候会有小坑,比如直接单独运行测试文件可能会报错,所以还是更推荐方法二。
额外小技巧:用pytest.ini省事儿
在项目根目录创建一个pytest.ini文件,内容如下:
[pytest] testpaths = test pythonpath = src
之后不管你在项目根目录怎么跑pytest,它都会自动把src加入到模块搜索路径里,测试文件里直接写from script2 import fun2就行,完全不用改导入代码!
内容的提问来源于stack exchange,提问作者Johnny Chiu
相关产品推荐
相关产品推荐

