improved testing

Chrisyhjiang · Jun 27, 2024 · d8a1d72 · d8a1d72
1 parent c2a54b5
commit d8a1d72
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/pkg/cli/new_test.go b/src/pkg/cli/new_test.go
@@ -2,6 +2,7 @@ package cli
 
 import (
 	"context"
+	"errors"
 	"testing"
 )
 
@@ -10,7 +11,7 @@ func TestInitFromSamples(t *testing.T) {
 	if err == nil {
 		t.Fatal("Expected test to fail")
 	}
-	if err.Error() != "sample not found" {
-		t.Error("Expected 'sample not found' error")
+	if !errors.Is(err, ErrSampleNotFound) {
+		t.Errorf("Expected error to be %v, got %v", ErrSampleNotFound, err)
 	}
 }