17 lines
559 B
Python
17 lines
559 B
Python
# -*- 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
|