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 time
import os import os
import shutil import shutil
import tempfile
import numpy as np import numpy as np
import pytest import pytest
from pathlib import Path 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("TEST 10: JsonKVStorage Workspace Isolation (Integration)")
print("=" * 60) print("=" * 60)
# Create temporary test directory # Create temporary test directory under project temp/
test_dir = tempfile.mkdtemp(prefix="lightrag_test_kv_") 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}") print(f"\n Using test directory: {test_dir}")
try: 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("TEST 11: LightRAG End-to-End Workspace Isolation")
print("=" * 60) print("=" * 60)
# Create temporary test directory # Create temporary test directory under project temp/
# test_dir = tempfile.mkdtemp(prefix="lightrag_test_e2e_") test_dir = str(
test_dir = str(Path(__file__).parent.parent / "temp/e2e_workspace_isolation") Path(__file__).parent.parent
/ "temp/test_lightrag_end_to_end_workspace_isolation"
)
if os.path.exists(test_dir): if os.path.exists(test_dir):
shutil.rmtree(test_dir) shutil.rmtree(test_dir)
os.makedirs(test_dir, exist_ok=True) os.makedirs(test_dir, exist_ok=True)