1 files changed, 60 insertions, 0 deletions
diff --git a/termui/linegraph_test.go b/termui/linegraph_test.go
new file mode 100644
index 0000000..e8f8a6a
--- /dev/null
+++ b/termui/linegraph_test.go
@@ -0,0 +1,60 @@
+package termui
+
+import "testing"
+import "sort"
+
+func TestLess(t *testing.T) {
+	tests := []struct {
+		a, b string
+		e    bool
+	}{
+		{a: "abc", b: "def", e: true},
+		{a: "abc", b: "abc", e: true},
+		{a: "def", b: "abc", e: false},
+		{a: "1", b: "10", e: true},
+		{a: "1", b: "2", e: true},
+		{a: "a2", b: "2", e: false},
+		{a: "a2", b: "a10", e: true},
+		{a: "a20", b: "a2", e: false},
+		{a: "abc20", b: "def2", e: true},
+		{a: "abc20", b: "abc2", e: false},
+		{a: "abc20", b: "abc20", e: true},
+		{a: "abc30", b: "abc20", e: false},
+		{a: "abc20a", b: "abc20", e: false},
+		{a: "abc20", b: "abc20a", e: true},
+		{a: "abc20", b: "abc2a", e: false},
+		{a: "abc20", b: "abc3a", e: false},
+		{a: "abc20", b: "abc2abc", e: false},
+	}
+	for _, k := range tests {
+		n := numbered([]string{k.a, k.b})
+		g := n.Less(0, 1)
+		if g != k.e {
+			t.Errorf("%s < %s: expected %v, got %v", k.a, k.b, k.e, g)
+		}
+	}
+}
+
+func TestSort(t *testing.T) {
+	tests := []struct {
+		in, ex numbered
+	}{
+		{
+			in: numbered{"abc", "def", "abc", "abc", "def", "abc", "1", "10", "1", "2", "a2", "2", "a2", "a10", "a20", "a2", "abc20", "def2", "abc20", "abc2", "abc20", "abc20", "abc30", "abc20", "abc20a", "abc20", "abc20", "abc20a", "abc20", "abc2a"},
+			ex: numbered{"1", "1", "2", "2", "10", "a2", "a2", "a2", "a10", "a20", "abc", "abc", "abc", "abc", "abc2", "abc2a", "abc20", "abc20", "abc20", "abc20", "abc20", "abc20", "abc20", "abc20", "abc20a", "abc20a", "abc30", "def", "def", "def2"},
+		},
+		{
+			in: numbered{"CPU12", "CPU11", "CPU9", "CPU3", "CPU4", "CPU0", "CPU6", "CPU7", "CPU8", "CPU5", "CPU10", "CPU1", "CPU2"},
+			ex: numbered{"CPU0", "CPU1", "CPU2", "CPU3", "CPU4", "CPU5", "CPU6", "CPU7", "CPU8", "CPU9", "CPU10", "CPU11", "CPU12"},
+		},
+	}
+
+	for _, k := range tests {
+		sort.Sort(k.in)
+		for i, v := range k.in {
+			if v != k.ex[i] {
+				t.Errorf("failed to properly sort\n\texpected: %v\n\tgot:      %v", k.ex, k.in)
+			}
+		}
+	}
+}