lint: Fix testifylint

sue445 · Jul 13, 2024 · ad4ef3b · ad4ef3b
1 parent af21686
commit ad4ef3b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/db/undirected_graph_test.go b/db/undirected_graph_test.go
@@ -9,8 +9,8 @@ func TestUndirectedGraph_PutSymmetric(t *testing.T) {
 	g := NewUndirectedGraph()
 	g.PutSymmetric("a", "b", true)
 
-	assert.Equal(t, true, g.matrix["a"]["b"])
-	assert.Equal(t, true, g.matrix["b"]["a"])
+	assert.True(t, g.matrix["a"]["b"])
+	assert.True(t, g.matrix["b"]["a"])
 }
 
 func TestUndirectedGraph_GetRowColumns(t *testing.T) {

diff --git a/lib/erd_generator_test.go b/lib/erd_generator_test.go
@@ -111,7 +111,7 @@ func TestErdGenerator_generatePlantUmlErd(t *testing.T) {
 				Distance: tt.fields.Distance,
 			}
 			got := g.generatePlantUmlErd(tt.args.schema)
-			assert.Greater(t, len(got), 0)
+			assert.NotEmpty(t, got)
 		})
 	}
 }
@@ -201,7 +201,7 @@ func TestErdGenerator_generateMermaidErd(t *testing.T) {
 				Distance: tt.fields.Distance,
 			}
 			got := g.generateMermaidErd(tt.args.schema)
-			assert.Greater(t, len(got), 0)
+			assert.NotEmpty(t, got)
 		})
 	}
 }
@@ -326,7 +326,7 @@ func TestErdGenerator_generate_withSkipTable(t *testing.T) {
 			got, err := g.generate(tt.args.schema)
 			if assert.NoError(t, err) {
 				if len(tt.wantContainTables) == 0 {
-					assert.Equal(t, got, "")
+					assert.Equal(t, "", got)
 				} else {
 					for _, tableName := range tt.wantContainTables {
 						assert.Contains(t, got, tableName)