Files
iAOP/core/rag-kb/tests/_bootstrap.py
T

17 lines
559 B
Python
Raw Normal View History

# -*- coding: utf-8 -*-
"""测试引导:把 `core/rag-kb` 以包名 `rag_kb` 挂载到 sys.modules。
目录名 `rag-kb` 含连字符,无法直接以包名 import;挂载后模块内相对导入
(`from .templating import ...`)在 unittest 发现机制下可正常解析。
"""
import os
import sys
import types
RAG_KB_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, RAG_KB_DIR)
if "rag_kb" not in sys.modules:
pkg = types.ModuleType("rag_kb")
pkg.__path__ = [RAG_KB_DIR]
sys.modules["rag_kb"] = pkg