diff options
Diffstat (limited to 'go/mixed-server_test.go')
-rw-r--r-- | go/mixed-server_test.go | 24 |
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) + } +} |