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
|
package dstream
import (
"math"
"testing"
)
func describeData() Dstream {
x := [][]interface{}{
{
[]float64{0, 0, 0},
[]float64{1, 1, 1},
[]float64{2, 2, 2},
},
{
[]float64{1, 3, 5},
[]float64{1, 3, 5},
[]float64{1, 3, 5},
},
{
[]float64{5, 5, 5},
[]float64{5, math.NaN(), 5},
[]float64{5, math.Inf(-1), math.Inf(1)},
},
}
na := []string{"x1", "x2", "x3"}
return NewFromArrays(x, na)
}
func compareStats(a, b Stats) bool {
if math.Abs(a.Mean-b.Mean) > 1e-6 {
return false
}
if math.Abs(a.Min-b.Min) > 1e-6 {
return false
}
if math.Abs(a.Max-b.Max) > 1e-6 {
return false
}
if math.Abs(a.SD-b.SD) > 1e-6 {
return false
}
return true
}
func TestDescribe1(t *testing.T) {
da := describeData()
st := Describe(da)
// Correct values
e := map[string]Stats{
"x1": Stats{
Min: 0,
Max: 2,
Mean: 1,
SD: math.Sqrt(6.0 / 9.0),
N: 9,
NaN: 0,
Inf: 0,
},
"x2": Stats{
Min: 1,
Max: 5,
Mean: 3,
SD: math.Sqrt(24.0 / 9.0),
N: 9,
NaN: 0,
Inf: 0,
},
"x3": Stats{
Min: 5,
Max: 5,
Mean: 5,
SD: 0,
N: 6,
Inf: 2,
NaN: 1,
},
}
for k, v := range e {
if !compareStats(v, st[k]) {
t.Fail()
}
}
}
|
