alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-21 21:22:11 -03:00
Code Issues Packages Projects Releases Wiki Activity

Skip performance tests by default

Browse Source 
- Add 'performance' marker to pytest.ini
- Add pytestmark to test_cache_performance.py
- Use -m 'not performance' by default in addopts
- Allows manual execution with 'pytest -m performance'
This commit is contained in:
Will Miao
2026-02-28 21:46:20 +08:00
parent c9e5ea42cb
commit 24fcbeaf76
2 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pytest.ini
View File

@@ -1,5 +1,5 @@
[pytest] [pytest]
addopts = -v --import-mode=importlib addopts = -v --import-mode=importlib -m "not performance"
testpaths = tests testpaths = tests
python_files = test_*.py python_files = test_*.py
python_classes = Test* python_classes = Test*
@@ -12,5 +12,6 @@ markers =
asyncio: execute test within asyncio event loop asyncio: execute test within asyncio event loop
no_settings_dir_isolation: allow tests to use real settings paths no_settings_dir_isolation: allow tests to use real settings paths
integration: integration tests requiring external resources integration: integration tests requiring external resources
performance: performance benchmarks (slow, skip by default)
# Skip problematic directories to avoid import conflicts # Skip problematic directories to avoid import conflicts
norecursedirs = .git .tox dist build *.egg __pycache__ py .hypothesis norecursedirs = .git .tox dist build *.egg __pycache__ py .hypothesis

21
tests/performance/test_cache_performance.py
View File

@@ -14,12 +14,17 @@ from py.services.model_hash_index import ModelHashIndex
from py.utils.utils import fuzzy_match, calculate_recipe_fingerprint from py.utils.utils import fuzzy_match, calculate_recipe_fingerprint
pytestmark = pytest.mark.performance
class TestHashIndexPerformance: class TestHashIndexPerformance:
"""Performance benchmarks for hash index operations.""" """Performance benchmarks for hash index operations."""
def test_hash_index_lookup_small(self, benchmark): def test_hash_index_lookup_small(self, benchmark):
"""Benchmark hash index lookup with 100 models.""" """Benchmark hash index lookup with 100 models."""
index, target_hash = self._create_hash_index_with_n_models(100, return_target=True) index, target_hash = self._create_hash_index_with_n_models(
100, return_target=True
)
def lookup(): def lookup():
return index.get_path(target_hash) return index.get_path(target_hash)
@@ -29,7 +34,9 @@ class TestHashIndexPerformance:
def test_hash_index_lookup_medium(self, benchmark): def test_hash_index_lookup_medium(self, benchmark):
"""Benchmark hash index lookup with 1,000 models.""" """Benchmark hash index lookup with 1,000 models."""
index, target_hash = self._create_hash_index_with_n_models(1000, return_target=True) index, target_hash = self._create_hash_index_with_n_models(
1000, return_target=True
)
def lookup(): def lookup():
return index.get_path(target_hash) return index.get_path(target_hash)
@@ -39,7 +46,9 @@ class TestHashIndexPerformance:
def test_hash_index_lookup_large(self, benchmark): def test_hash_index_lookup_large(self, benchmark):
"""Benchmark hash index lookup with 10,000 models.""" """Benchmark hash index lookup with 10,000 models."""
index, target_hash = self._create_hash_index_with_n_models(10000, return_target=True) index, target_hash = self._create_hash_index_with_n_models(
10000, return_target=True
)
def lookup(): def lookup():
return index.get_path(target_hash) return index.get_path(target_hash)
@@ -71,11 +80,11 @@ class TestHashIndexPerformance:
def _create_hash_index_with_n_models(self, n: int, return_target: bool = False): def _create_hash_index_with_n_models(self, n: int, return_target: bool = False):
"""Create a hash index with n mock models. """Create a hash index with n mock models.
Args: Args:
n: Number of models to create n: Number of models to create
return_target: If True, returns the hash of the middle model for lookup testing return_target: If True, returns the hash of the middle model for lookup testing
Returns: Returns:
ModelHashIndex or tuple of (ModelHashIndex, target_hash) ModelHashIndex or tuple of (ModelHashIndex, target_hash)
""" """
@@ -94,7 +103,7 @@ class TestHashIndexPerformance:
def _random_string(self, length: int) -> str: def _random_string(self, length: int) -> str:
"""Generate a random string of fixed length.""" """Generate a random string of fixed length."""
return ''.join(random.choices(string.ascii_lowercase + string.digits, k=length)) return "".join(random.choices(string.ascii_lowercase + string.digits, k=length))
class TestFuzzyMatchPerformance: class TestFuzzyMatchPerformance: