Fix empty search case

Signed-off-by: Congqi Xia <[email protected]>
milvus-io · Apr 18, 2024 · 3267d02 · 3267d02
1 parent 5314569
commit 3267d02
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/test/testcases/search_test.go b/test/testcases/search_test.go
@@ -8,6 +8,7 @@ import (
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 
 	"github.com/milvus-io/milvus-sdk-go/v2/client"
@@ -175,7 +176,10 @@ func TestSearchEmptyCollection(t *testing.T) {
 			common.DefaultTopK,
 			sp,
 		)
-		require.Len(t, searchRes, 0)
+		require.Len(t, searchRes, common.DefaultNq)
+		for _, resultSet := range searchRes {
+			assert.EqualValues(t, 0, resultSet.ResultCount)
+		}
 	}
 }
 
@@ -1151,7 +1155,10 @@ func TestSearchArrayFieldExpr(t *testing.T) {
 			sp,
 		)
 		common.CheckErr(t, errSearchEmpty, true)
-		require.Empty(t, searchRes)
+		require.Len(t, searchRes, common.DefaultNq)
+		for _, resultSet := range searchRes {
+			assert.EqualValues(t, 0, resultSet.ResultCount)
+		}
 	}
 }