gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Standardize test directory creation and remove tempfile dependency

Browse source 
• Remove unused tempfile import
• Use consistent project temp/ structure
• Clean up existing directories first
• Create directories with os.makedirs
• Use descriptive test directory names
This commit is contained in:
yangdx 2025-11-18 10:39:54 +08:00
parent 1fe05df211
commit 4fef731f37
1 changed files with 12 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/test_workspace_isolation.py
View file

@ -22,7 +22,6 @@ import asyncio
import time
import os
import shutil
import tempfile
import numpy as np
import pytest
from pathlib import Path
@ -786,8 +785,13 @@ async def test_json_kv_storage_workspace_isolation(keep_test_artifacts):
print("TEST 10: JsonKVStorage Workspace Isolation (Integration)")
print("=" * 60)
# Create temporary test directory
test_dir = tempfile.mkdtemp(prefix="lightrag_test_kv_")
# Create temporary test directory under project temp/
test_dir = str(
Path(__file__).parent.parent / "temp/test_json_kv_storage_workspace_isolation"
)
if os.path.exists(test_dir):
shutil.rmtree(test_dir)
os.makedirs(test_dir, exist_ok=True)
print(f"\n Using test directory: {test_dir}")
try:
@ -954,9 +958,11 @@ async def test_lightrag_end_to_end_workspace_isolation(keep_test_artifacts):
print("TEST 11: LightRAG End-to-End Workspace Isolation")
print("=" * 60)
# Create temporary test directory
# test_dir = tempfile.mkdtemp(prefix="lightrag_test_e2e_")
test_dir = str(Path(__file__).parent.parent / "temp/e2e_workspace_isolation")
# Create temporary test directory under project temp/
test_dir = str(
Path(__file__).parent.parent
/ "temp/test_lightrag_end_to_end_workspace_isolation"
)
if os.path.exists(test_dir):
shutil.rmtree(test_dir)
os.makedirs(test_dir, exist_ok=True)