python 框架代码中执行脚本单元测试时导入模块
在python框架项目开发中,一般都会用到脚本来进行单元测试,因此脚本也就可能需要调用其他模块(文件夹)下的功能代码,这时就需要将要调用的模块目录导入到测试脚本的环境变量中,导入方法可以使用sys.path.append
, 参考如下:
# test_xx.py
import sys
import os
curr_dir = os.path.dirname(os.path.abspath(__file__))
sys.path.append(os.path.join(curr_dir, "..")) # sys.path.append 将上级目录添加到脚本的环境变量中
sys.path.append(os.path.join(curr_dir, "..db"))
sys.path.append(os.path.join(curr_dir, "..model"))
sys.path.append(os.path.join(curr_dir, "..tools"))
sys.path.append(os.path.join(curr_dir, "..routers"))
from datetime import datetime
from db.db_proc import * # 调用 db/db_proc.py 的属性对象
from tools import get_format_time # 调用 tools/get_format_time.py 的属性对象