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
|
package dstream
import "testing"
func datal1() (Dstream, Dstream) {
x1 := []interface{}{
[]float64{0, 0, 0, 0, 0, 0},
[]float64{1, 1, 1, 2, 3, 4},
[]float64{2, 4, 6, 8, 10, 12},
}
x2 := []interface{}{
[]float64{0, 0, 1, 2, 3, 4},
[]float64{1, 2, 2, 2, 2, 1},
[]float64{3, 3, 3, 3, 3, 3},
}
x3 := []interface{}{
[]float64{1, 2, 3, 4, 5, 6},
[]float64{4, 5, 6, 7, 8, 9},
[]float64{7, 8, 9, 10, 11, 12},
}
x4 := []interface{}{
[]string{"a", "b", "c", "d", "e", "f"},
[]string{"d", "e", "f", "g", "h", "i"},
[]string{"g", "h", "i", "j", "k", "l"},
}
dat := [][]interface{}{x1, x2, x3, x4}
na := []string{"x1", "x2", "x3", "x4"}
da := NewFromArrays(dat, na)
x1_0 := []interface{}{
[]float64{0, 0, 0, 0},
[]float64{1, 2, 3, 4},
[]float64{6, 8, 10, 12},
}
x1_1 := []interface{}{
[]float64{0, 0, 0, 0},
[]float64{1, 1, 2, 3},
[]float64{4, 6, 8, 10},
}
x2_0 := []interface{}{
[]float64{1, 2, 3, 4},
[]float64{2, 2, 2, 1},
[]float64{3, 3, 3, 3},
}
x2_1 := []interface{}{
[]float64{0, 1, 2, 3},
[]float64{2, 2, 2, 2},
[]float64{3, 3, 3, 3},
}
x2_2 := []interface{}{
[]float64{0, 0, 1, 2},
[]float64{1, 2, 2, 2},
[]float64{3, 3, 3, 3},
}
x3 = []interface{}{
[]float64{3, 4, 5, 6},
[]float64{6, 7, 8, 9},
[]float64{9, 10, 11, 12},
}
x4 = []interface{}{
[]string{"c", "d", "e", "f"},
[]string{"f", "g", "h", "i"},
[]string{"i", "j", "k", "l"},
}
dat = [][]interface{}{x1_0, x1_1, x2_0, x2_1, x2_2, x3, x4}
na = []string{"x1[0]", "x1[-1]", "x2[0]", "x2[-1]", "x2[-2]", "x3", "x4"}
dm := NewFromArrays(dat, na)
return da, dm
}
func TestLag1(t *testing.T) {
da, dm := datal1()
dx := LagChunk(da, map[string]int{"x1": 1, "x2": 2})
if !Equal(dm, dx) {
t.Fail()
}
}
|
