コード例 #1
0
ファイル: spacesaving_test.go プロジェクト: timwee/goru
func TestSpaceSavingSanity(t *testing.T) {
	ss := topk.MakeSpaceSaving(4)
	for i := 0; i < 10; i++ {
		ss.Observe("a")
	}
	c := ss.GetCounter("a")
	if c.GetCount() != 10 {
		t.Errorf("expected 'a' to have count of 10, but was %i", c.GetCount())
	}
}
コード例 #2
0
ファイル: spacesaving_test.go プロジェクト: timwee/goru
func TestTopK(t *testing.T) {
	ss := topk.MakeSpaceSaving(4)
	xs := []string{"a", "b", "c", "d", "a", "b", "e", "a", "b"}
	for _, s := range xs {
		ss.Observe(s)
	}
	k := 2
	topk, _ := ss.TopK(int32(k))
	if len(topk) != 2 {
		t.Errorf("result didn't have %d elements, instead was %d", k, len(topk))
	}

	for _, c := range topk {
		if c.Key != "a" && c.Key != "b" {
			t.Errorf("key of top2 not 'a' or 'b' was %s", c.Key)
		}
	}
}