fix: fix test case TestGraph

Signed-off-by: zongz <[email protected]>
kcl-lang · Nov 13, 2024 · 6e95e1d · 6e95e1d
1 parent ad1318d
commit 6e95e1d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pkg/client/graph_test.go b/pkg/client/graph_test.go
@@ -4,8 +4,8 @@ import (
 	"path/filepath"
 	"testing"
 
+	"github.com/stretchr/testify/assert"
 	"golang.org/x/mod/module"
-	"gotest.tools/v3/assert"
 	pkg "kcl-lang.io/kpm/pkg/package"
 	"kcl-lang.io/kpm/pkg/utils"
 )
@@ -43,5 +43,7 @@ func testGraph(t *testing.T) {
 		t.Fatalf("failed to display graph: %v", err)
 	}
 
-	assert.Equal(t, utils.RmNewline(graStr), "[email protected] [email protected]@0.0.1 [email protected]@0.0.1 [email protected]")
+	assert.Contains(t, utils.RmNewline(graStr), "[email protected] [email protected]")
+	assert.Contains(t, utils.RmNewline(graStr), "[email protected] [email protected]")
+	assert.Contains(t, utils.RmNewline(graStr), "[email protected] [email protected]")
 }