fix kb system test

pinecone-io · Oct 2, 2023 · 995e639 · 995e639
1 parent 839d738
commit 995e639
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/system/knowledge_base/test_knowledge_base.py b/tests/system/knowledge_base/test_knowledge_base.py
@@ -206,9 +206,7 @@ def test_upsert_dataframe(knowledge_base, documents, chunker, encoder):
     for chunk in encoded_chunks:
         chunk.source = ''
 
-    vec_after = total_vectors_in_index(knowledge_base)
-
-    assert vec_after - vec_before == len(encoded_chunks)
+    assert_num_vectors_in_index(knowledge_base, vec_before + len(encoded_chunks))
     assert_chunks_in_index(knowledge_base, encoded_chunks)