Skip to content

Commit 92f7a6b

Browse files
committed
refactor(tests): streamline test client add method calls for clarity
- Simplified the `add` method calls in the `TestUniqueText` class by removing unnecessary line breaks for better readability. - Ensured consistency in the formatting of test cases related to handling unique text entries.
1 parent cbe9f4f commit 92f7a6b

1 file changed

Lines changed: 5 additions & 15 deletions

File tree

tests/test_client.py

Lines changed: 5 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -104,9 +104,7 @@ def test_unique_text_rejects_duplicates_by_default(
104104
self, client_with_unique_table, sample_embeddings
105105
):
106106
"""Duplicate text raises IntegrityError when on_conflict='error'."""
107-
client_with_unique_table.add(
108-
texts=["hello"], embeddings=[sample_embeddings[0]]
109-
)
107+
client_with_unique_table.add(texts=["hello"], embeddings=[sample_embeddings[0]])
110108
with pytest.raises(sqlite3.IntegrityError):
111109
client_with_unique_table.add(
112110
texts=["hello"], embeddings=[sample_embeddings[1]]
@@ -116,9 +114,7 @@ def test_on_conflict_ignore_skips_duplicates(
116114
self, client_with_unique_table, sample_embeddings
117115
):
118116
"""Duplicate texts are silently skipped with on_conflict='ignore'."""
119-
client_with_unique_table.add(
120-
texts=["hello"], embeddings=[sample_embeddings[0]]
121-
)
117+
client_with_unique_table.add(texts=["hello"], embeddings=[sample_embeddings[0]])
122118
rowids = client_with_unique_table.add(
123119
texts=["hello", "world"],
124120
embeddings=[sample_embeddings[1], sample_embeddings[2]],
@@ -131,9 +127,7 @@ def test_on_conflict_ignore_all_duplicates(
131127
self, client_with_unique_table, sample_embeddings
132128
):
133129
"""All duplicates skipped returns empty rowids."""
134-
client_with_unique_table.add(
135-
texts=["hello"], embeddings=[sample_embeddings[0]]
136-
)
130+
client_with_unique_table.add(texts=["hello"], embeddings=[sample_embeddings[0]])
137131
rowids = client_with_unique_table.add(
138132
texts=["hello"],
139133
embeddings=[sample_embeddings[1]],
@@ -185,9 +179,7 @@ def test_on_conflict_replace_mixed_insert_and_update(
185179
self, client_with_unique_table, sample_embeddings
186180
):
187181
"""Replace mode handles a mix of new and existing texts."""
188-
client_with_unique_table.add(
189-
texts=["hello"], embeddings=[sample_embeddings[0]]
190-
)
182+
client_with_unique_table.add(texts=["hello"], embeddings=[sample_embeddings[0]])
191183
rowids = client_with_unique_table.add(
192184
texts=["hello", "world"],
193185
embeddings=[sample_embeddings[1], sample_embeddings[2]],
@@ -214,9 +206,7 @@ def test_on_conflict_replace_vec_table_synced(
214206
self, client_with_unique_table, sample_embeddings
215207
):
216208
"""Replace mode keeps the vector table in sync for similarity search."""
217-
client_with_unique_table.add(
218-
texts=["hello"], embeddings=[sample_embeddings[0]]
219-
)
209+
client_with_unique_table.add(texts=["hello"], embeddings=[sample_embeddings[0]])
220210
client_with_unique_table.add(
221211
texts=["hello"],
222212
embeddings=[sample_embeddings[1]],

0 commit comments

Comments
 (0)