File: mutate_test.go

package info (click to toggle)
golang-github-kshedden-dstream 0.0~git20190512.c4c4106-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 596 kB
  • sloc: makefile: 30
file content (47 lines) | stat: -rw-r--r-- 831 bytes parent folder | download
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
package dstream

import "testing"

func TestMutate1(t *testing.T) {

	x1 := []interface{}{
		[]float64{0, 0, 0},
		[]float64{1, 1, 1},
		[]float64{2, 2, 3},
	}
	x2 := []interface{}{
		[]string{"a", "b", "c"},
		[]string{"d", "e", "f"},
		[]string{"g", "h", "i"},
	}
	x3 := []interface{}{
		[]float64{1, 2, 3},
		[]float64{4, 5, 6},
		[]float64{7, 8, 9},
	}
	dat := [][]interface{}{x1, x2, x3}
	na := []string{"x1", "x2", "x3"}
	da := NewFromArrays(dat, na)

	x3t := []interface{}{
		[]float64{3, 5, 7},
		[]float64{9, 11, 13},
		[]float64{15, 17, 19},
	}
	dat = [][]interface{}{x1, x2, x3t}
	na = []string{"x1", "x2", "x3"}
	de := NewFromArrays(dat, na)

	f := func(x interface{}) {
		z := x.([]float64)
		for i, y := range z {
			z[i] = 2*y + 1
		}
	}

	dm := Mutate(da, "x3", f)

	if !EqualReport(dm, de, true) {
		t.Fail()
	}
}