go model

1 files changed, 24 insertions, 0 deletions

diff --git a/go/mixed-server_test.go b/go/mixed-server_test.go
new file mode 100644
index 0000000..6e33b1f
--- /dev/null
+++ b/go/mixed-server_test.go
@@ -0,0 +1,24 @@
+package main
+
+import (
+	"testing"
+)
+
+func TestModel1(t *testing.T) {
+	m := NewModel()
+	m.AddWords([]string{"a", "b", "a"})
+	got := m.GetMostCommonPairs(2)
+	if !(got[0] == "a" && got[1] == "b") {
+		t.Errorf("bad order, got %v", got)
+	}
+}
+
+func TestModel2(t *testing.T) {
+	m := NewModel()
+	m.AddWords([]string{"a", "b", "a"})
+	m.AddWords([]string{"b", "c", "b"})
+	got := m.GetMostCommonPairs(2)
+	if !(got[0] == "b" && got[1] == "a") {
+		t.Errorf("bad order, got %v", got)
+	}
+}