fix checking indexes after cleanup

pinecone-io · Sep 20, 2024 · 2912ad7 · 2912ad7
1 parent 7e3913a
commit 2912ad7
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/pinecone/local_test.go b/pinecone/local_test.go
@@ -65,12 +65,12 @@ func (ts *LocalIntegrationTests) TearDownSuite() {
 		// Delete all remaining vectors
 		err = idxConn.DeleteAllVectorsInNamespace(context.Background())
 		require.NoError(ts.T(), err)
-
-		description, err := idxConn.DescribeIndexStats(context.Background())
-		require.NoError(ts.T(), err)
-		assert.NotNil(ts.T(), description, "Index description should not be nil")
-		assert.Equal(ts.T(), uint32(0), description.TotalVectorCount, "Total vector count should be 0 after deleting")
 	}
+
+	description, err := ts.idxConns[0].DescribeIndexStats(context.Background())
+	require.NoError(ts.T(), err)
+	assert.NotNil(ts.T(), description, "Index description should not be nil")
+	assert.Equal(ts.T(), uint32(0), description.TotalVectorCount, "Total vector count should be 0 after deleting")
 }
 
 // This is the entry point for all local integration tests