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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
package stats_test
import (
"testing"
"github.com/montanaflynn/stats"
)
func TestMean(t *testing.T) {
for _, c := range []struct {
in []float64
out float64
}{
{[]float64{1, 2, 3, 4, 5}, 3.0},
{[]float64{1, 2, 3, 4, 5, 6}, 3.5},
{[]float64{1}, 1.0},
} {
got, _ := stats.Mean(c.in)
if got != c.out {
t.Errorf("Mean(%.1f) => %.1f != %.1f", c.in, got, c.out)
}
}
_, err := stats.Mean([]float64{})
if err == nil {
t.Errorf("Empty slice should have returned an error")
}
}
func BenchmarkMeanSmallFloatSlice(b *testing.B) {
for i := 0; i < b.N; i++ {
_, _ = stats.Mean(makeFloatSlice(5))
}
}
func BenchmarkMeanLargeFloatSlice(b *testing.B) {
lf := makeFloatSlice(100000)
b.ResetTimer()
for i := 0; i < b.N; i++ {
_, _ = stats.Mean(lf)
}
}
func TestGeometricMean(t *testing.T) {
s1 := []float64{2, 18}
s2 := []float64{10, 51.2, 8}
s3 := []float64{1, 3, 9, 27, 81}
for _, c := range []struct {
in []float64
out float64
}{
{s1, 6},
{s2, 16},
{s3, 9},
} {
gm, err := stats.GeometricMean(c.in)
if err != nil {
t.Errorf("Should not have returned an error")
}
gm, _ = stats.Round(gm, 0)
if gm != c.out {
t.Errorf("Geometric Mean %v != %v", gm, c.out)
}
}
_, err := stats.GeometricMean([]float64{})
if err == nil {
t.Errorf("Empty slice should have returned an error")
}
}
func TestHarmonicMean(t *testing.T) {
s1 := []float64{1, 2, 3, 4, 5}
s2 := []float64{10, -51.2, 8}
s3 := []float64{1, 0, 9, 27, 81}
hm, err := stats.HarmonicMean(s1)
if err != nil {
t.Errorf("Should not have returned an error")
}
hm, _ = stats.Round(hm, 2)
if hm != 2.19 {
t.Errorf("Geometric Mean %v != %v", hm, 2.19)
}
_, err = stats.HarmonicMean(s2)
if err == nil {
t.Errorf("Should have returned a negative number error")
}
_, err = stats.HarmonicMean(s3)
if err == nil {
t.Errorf("Should have returned a zero number error")
}
_, err = stats.HarmonicMean([]float64{})
if err == nil {
t.Errorf("Empty slice should have returned an error")
}
}
|
