TBNilles/LEANN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: format test files with latest ruff version for CI compatibility

Browse Source
This commit is contained in:
Andy Lee
2025-08-06 22:53:40 -07:00
parent 9842ad8330
commit 6061e8f2de
2 changed files with 27 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
tests/test_diskann_partition.py
View File

@@ -71,9 +71,9 @@ def test_diskann_without_partition():
for partition_file in partition_files:
file_path = index_dir / partition_file
assert (
not file_path.exists()
), f"Partition file {partition_file} should not exist in non-partition mode"
assert not file_path.exists(), (
f"Partition file {partition_file} should not exist in non-partition mode"
)
# Test search functionality
searcher = LeannSearcher(index_path)
@@ -137,9 +137,9 @@ def test_diskann_with_partition():
for large_file in large_files:
file_path = index_dir / large_file
assert (
not file_path.exists()
), f"Large file {large_file} should have been deleted for storage saving"
assert not file_path.exists(), (
f"Large file {large_file} should have been deleted for storage saving"
)
# Verify required auxiliary files for partition mode exist
required_files = [
@@ -149,9 +149,9 @@ def test_diskann_with_partition():
for req_file in required_files:
file_path = index_dir / req_file
assert (
file_path.exists()
), f"Required auxiliary file {req_file} missing for partition mode"
assert file_path.exists(), (
f"Required auxiliary file {req_file} missing for partition mode"
)
@pytest.mark.skipif(
@@ -203,21 +203,21 @@ def test_diskann_partition_search_functionality():
# Verify search results
assert len(results) == top_k, f"Expected {top_k} results for query '{query}'"
assert all(
result.score is not None for result in results
), "All results should have scores"
assert all(
result.score != float("-inf") for result in results
), "No result should have -inf score"
assert all(
result.text is not None for result in results
), "All results should have text"
assert all(result.score is not None for result in results), (
"All results should have scores"
)
assert all(result.score != float("-inf") for result in results), (
"No result should have -inf score"
)
assert all(result.text is not None for result in results), (
"All results should have text"
)
# Scores should be in descending order (higher similarity first)
scores = [result.score for result in results]
assert scores == sorted(
scores, reverse=True
), "Results should be sorted by score descending"
assert scores == sorted(scores, reverse=True), (
"Results should be sorted by score descending"
)
@pytest.mark.skipif(
@@ -286,9 +286,9 @@ def test_diskann_medoid_and_norm_files():
# Verify that scores are not -inf (which indicates norm file was loaded correctly)
assert len(results) > 0
assert all(
result.score != float("-inf") for result in results
), "Scores should not be -inf when norm file is correct"
assert all(result.score != float("-inf") for result in results), (
"Scores should not be -inf when norm file is correct"
)
@pytest.mark.skipif(

6
tests/test_readme_examples.py
View File

@@ -53,9 +53,9 @@ def test_readme_basic_example(backend_name):
# Verify search results
assert len(results) > 0
assert isinstance(results[0], SearchResult)
assert results[0].score != float(
"-inf"
), f"should return valid scores, got {results[0].score}"
assert results[0].score != float("-inf"), (
f"should return valid scores, got {results[0].score}"
)
# The second text about banana-crocodile should be more relevant
assert "banana" in results[0].text or "crocodile" in results[0].text