fix tests

Author: shivaji-kharse

systest/vector/backup_test.go

@@ -41,16 +41,15 @@ func (vsuite *VectorTestSuite) TestVectorIncrBackupRestore() {
 	require.NoError(t, hc.LoginIntoNamespace(dgraphapi.DefaultUser,
 		dgraphapi.DefaultPassword, x.RootNamespace))
 
-	require.NoError(t, gc.SetupSchema(vsuite.schema))
+	require.NoError(t, gc.SetupSchema(vsuite.schemaVecDimesion10))
 
-	numVectors := 500
-	pred := "project_description_v"
+	numVectors := 1000
 	allVectors := make([][][]float32, 0, 5)
 	allRdfs := make([]string, 0, 5)
 	for i := 1; i <= 5; i++ {
 		var rdfs string
 		var vectors [][]float32
-		rdfs, vectors = dgraphapi.GenerateRandomVectors(numVectors*(i-1), numVectors*i, 1, pred)
+		rdfs, vectors = dgraphapi.GenerateRandomVectors(numVectors*(i-1), numVectors*i, 10, pred)
 		allVectors = append(allVectors, vectors)
 		allRdfs = append(allRdfs, rdfs)
 		mu := &api.Mutation{SetNquads: []byte(rdfs), CommitNow: true}
@@ -162,9 +161,9 @@ func (vsuite *VectorTestSuite) TestVectorBackupRestoreDropIndex() {
 	// add vector predicate + index
 	require.NoError(t, gc.SetupSchema(vsuite.schema))
 	// add data to the vector predicate
-	numVectors := 3
+	numVectors := 1000
 	pred := "project_description_v"
-	rdfs, vectors := dgraphapi.GenerateRandomVectors(0, numVectors, 1, pred)
+	rdfs, vectors := dgraphapi.GenerateRandomVectors(0, numVectors, 100, pred)
 	mu := &api.Mutation{SetNquads: []byte(rdfs), CommitNow: true}
 	_, err = gc.Mutate(mu)
 	require.NoError(t, err)
@@ -176,7 +175,7 @@ func (vsuite *VectorTestSuite) TestVectorBackupRestoreDropIndex() {
 	require.NoError(t, gc.SetupSchema(testSchemaWithoutIndex))
 
 	// add more data to the vector predicate
-	rdfs, vectors2 := dgraphapi.GenerateRandomVectors(3, numVectors+3, 1, pred)
+	rdfs, vectors2 := dgraphapi.GenerateRandomVectors(numVectors, numVectors+3, 100, pred)
 	mu = &api.Mutation{SetNquads: []byte(rdfs), CommitNow: true}
 	_, err = gc.Mutate(mu)
 	require.NoError(t, err)
@@ -214,7 +213,7 @@ func (vsuite *VectorTestSuite) TestVectorBackupRestoreDropIndex() {
 		}`
 	resp, err := gc.Query(query)
 	require.NoError(t, err)
-	require.JSONEq(t, `{"vectors":[{"count":4}]}`, string(resp.GetJson()))
+	require.JSONEq(t, `{"vectors":[{"count":1001}]}`, string(resp.GetJson()))
 
 	require.NoError(t, err)
 	allVec := append(vectors, vectors2...)

systest/vector/load_test.go

@@ -35,7 +35,7 @@ func (vsuite *VectorTestSuite) TestLiveLoadAndExportRDFFormat() {
 	defer func() { c.Cleanup(t.Failed()) }()
 	require.NoError(t, c.Start())
 
-	testExportAndLiveLoad(t, c, "rdf", vsuite.schema)
+	testExportAndLiveLoad(t, c, "rdf", vsuite.schemaVecDimesion10)
 }
 
 func testExportAndLiveLoad(t *testing.T, c *dgraphtest.LocalCluster, exportFormat string, schema string) {
@@ -54,7 +54,7 @@ func testExportAndLiveLoad(t *testing.T, c *dgraphtest.LocalCluster, exportForma
 
 	numVectors := 1000
 	pred := "project_description_v"
-	rdfs, vectors := dgraphapi.GenerateRandomVectors(0, numVectors, 100, pred)
+	rdfs, vectors := dgraphapi.GenerateRandomVectors(0, numVectors, 10, pred)
 
 	mu := &api.Mutation{SetNquads: []byte(rdfs), CommitNow: true}
 	_, err = gc.Mutate(mu)

systest/vector/vector_test.go

@@ -27,6 +27,7 @@ import (
 const (
 	testSchemaWithoutIndex = `project_description_v: float32vector .`
 	pred                   = "project_description_v"
+	schemaVecDimension10   = `project_description_v: float32vector @index(partionedhnsw(numClusters: "1000", partitionStratOpt: "kmeans", vectorDimension: "10", metric: "euclidean")) .`
 )
 
 var schemas = map[string]string{
@@ -619,6 +620,7 @@ func (vsuite *VectorTestSuite) TestPartitionedHNSWIndex() {
 type VectorTestSuite struct {
 	suite.Suite
 	schema                string
+	schemaVecDimesion10   string
 	isForPartitionedIndex bool
 }
 
@@ -627,7 +629,10 @@ func TestVectorSuite(t *testing.T) {
 		var ssuite VectorTestSuite
 		ssuite.schema = schema
 		if strings.Contains(schema, "partionedhnsw") {
+			ssuite.schemaVecDimesion10 = schemaVecDimension10
 			ssuite.isForPartitionedIndex = true
+		} else {
+			ssuite.schemaVecDimesion10 = schema
 		}
 		suite.Run(t, &ssuite)
 		if t.Failed() {

t/t.go

@@ -343,7 +343,7 @@ func sanitizeFilename(pkg string) string {
 
 func runTestsFor(ctx context.Context, pkg, prefix string, xmlFile string) error {
 	args := []string{"gotestsum", "--junitfile", xmlFile, "--format", "standard-verbose", "--max-fails", "1", "--",
-		"-v", "-failfast", "-tags=integration"}
+		"-v", "-tags=integration"}
 	if *race {
 		args = append(args, "-timeout", "180m")
 		// Todo: There are few race errors in tests itself. Enable this once that is fixed.