Add test for classify/labels

internal/classify/labels_test.go

@@ -55,4 +55,39 @@ func TestLabels_Title(t *testing.T) {
 
 		assert.Equal(t, "fallback", labels.Title("fallback"))
 	})
+
+	t.Run("empty labels", func(t *testing.T) {
+		labels := Labels{}
+
+		assert.Equal(t, "", labels.Title(""))
+	})
+
+	t.Run("label priority < 0", func(t *testing.T) {
+		cat := Label{Name: "cat", Source: "location", Uncertainty: 59, Priority: -1}
+		dog := Label{Name: "dog", Source: "location", Uncertainty: 10, Priority: -1}
+		labels := Labels{cat, dog}
+
+		assert.Equal(t, "fallback", labels.Title("fallback"))
+	})
+
+	t.Run("label priority = 0", func(t *testing.T) {
+		cat := Label{Name: "cat", Source: "location", Uncertainty: 59, Priority: 0}
+		dog := Label{Name: "dog", Source: "location", Uncertainty: 62, Priority: 0}
+		labels := Labels{cat, dog}
+
+		assert.Equal(t, "fallback", labels.Title("fallback"))
+	})
+}
+
+func TestLabels_Keywords(t *testing.T) {
+	cat := Label{Name: "cat", Source: "location", Uncertainty: 80, Priority: 5, Categories: []string{"animal"}}
+	dog := Label{Name: "dog", Source: "location", Uncertainty: 80, Priority: 5}
+	labels := Labels{cat, dog}
+
+	t.Run("labelWithName", func(t *testing.T) {
+		result := labels.Keywords()
+		assert.Equal(t, "cat", result[0])
+		assert.Equal(t, "animal", result[1])
+		assert.Equal(t, "dog", result[2])
+	})
 }