diff --git a/wtf/colors_test.go b/wtf/colors_test.go
index 595bee61..7ac90839 100644
--- a/wtf/colors_test.go
+++ b/wtf/colors_test.go
@@ -3,11 +3,73 @@ package wtf
 import (
 	"testing"
 
-	. "github.com/stretchr/testify/assert"
+	"github.com/gdamore/tcell"
 )
 
 func Test_ASCIItoTviewColors(t *testing.T) {
-	Equal(t, "", ASCIItoTviewColors(""))
-	Equal(t, "cat", ASCIItoTviewColors("cat"))
-	Equal(t, "[38;5;226mcat/[-]", ASCIItoTviewColors("[38;5;226mcat/"))
+	tests := []struct {
+		name     string
+		text     string
+		expected string
+	}{
+		{
+			name:     "with blank text",
+			text:     "",
+			expected: "",
+		},
+		{
+			name:     "with no color",
+			text:     "cat",
+			expected: "cat",
+		},
+		{
+			name: "with defined color",
+			text: "[38;5;226mcat/",
+			expected: "[38;5;226mcat/[-]",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			actual := ASCIItoTviewColors(tt.text)
+
+			if tt.expected != actual {
+				t.Errorf("\nexpected: %q\n     got: %q", tt.expected, actual)
+			}
+		})
+	}
+}
+
+func Test_ColorFor(t *testing.T) {
+	tests := []struct {
+		name     string
+		label    string
+		expected tcell.Color
+	}{
+		{
+			name:     "with no label",
+			label:    "",
+			expected: tcell.ColorGreen,
+		},
+		{
+			name:     "with missing label",
+			label:    "cat",
+			expected: tcell.ColorGreen,
+		},
+		{
+			name:     "with defined label",
+			label:    "tomato",
+			expected: tcell.ColorTomato,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			actual := ColorFor(tt.label)
+
+			if tt.expected != actual {
+				t.Errorf("\nexpected: %q\n     got: %q", tt.expected, actual)
+			}
+		})
+	}
 }