aboutsummaryrefslogtreecommitdiff
path: root/sketch_test.go
blob: 962dce09662c30b0ff87765117fd81e4da76ff7e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package main

import (
	"path"
	"reflect"
	"testing"
)

func TestSketchChunk(t *testing.T) {
	dataDir := path.Join("test", "data", "repo_8k")
	chunks := make(chan StoredChunk, 16)
	repo := NewRepo(dataDir)
	versions := repo.loadVersions()
	go repo.loadChunks(versions, chunks)
	var i int
	for c := range chunks {
		if i < 1 {
			sketch, err := SketchChunk(c, 8<<10, 32, 3, 4)
			if err != nil {
				t.Error(err)
			}
			expected := Sketch{429857165471867, 6595034117354675, 8697818304802825}
			if !reflect.DeepEqual(sketch, expected) {
				t.Errorf("Sketch does not match, expected: %d, actual: %d", expected, sketch)
			}
		}
		if i == 14 {
			sketch, err := SketchChunk(c, 8<<10, 32, 3, 4)
			if err != nil {
				t.Error(err)
			}
			expected := Sketch{658454504014104}
			if !reflect.DeepEqual(sketch, expected) {
				t.Errorf("Sketch does not match, expected: %d, actual: %d", expected, sketch)
			}
		}
		i++
	}
}