1 // Copyright ©2016 The Gonum Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style
3 // license that can be found in the LICENSE file.
10 incX, incY, incDst int
16 {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
21 ex: []complex64{1 + 2i}},
22 {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
24 dst: []complex64{0, 0, 0},
25 x: []complex64{0, 0, 0},
26 y: []complex64{1, 1, 1},
27 ex: []complex64{1, 1, 1}},
28 {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
30 dst: []complex64{0, 0, 0},
32 y: []complex64{1, 1, 1},
33 ex: []complex64{1, 1}},
34 {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
36 dst: []complex64{1i, 1i, 1i},
37 x: []complex64{1i, 1i, 1i},
38 y: []complex64{1, 2, 1},
39 ex: []complex64{-1 + 1i, 1i, -1 + 1i}},
40 {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
42 dst: []complex64{1i, 1i, 1i},
43 x: []complex64{1i, 1i, 1i},
44 y: []complex64{1, 2, 1},
45 ex: []complex64{2, 3, 2}},
46 {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
48 dst: []complex64{1i, 1i, 1i},
49 x: []complex64{1i, 1i, 1i, 1i, 1i}[1:4],
50 y: []complex64{1, 1, 2, 1, 1}[1:4],
51 ex: []complex64{2, 3, 2}},
52 {incX: 2, incY: 4, incDst: 3, ix: 0, iy: 0, idst: 0,
54 dst: []complex64{1i, 1i, 1i, 1i, 1i},
55 x: []complex64{2 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 2 + 1i},
56 y: []complex64{1, 1, 2, 1, 1},
57 ex: []complex64{-3 - 2i, -3 - 2i, -2 - 2i, -3 - 2i, -3 - 2i}},
58 // Run big test twice, once aligned once unaligned.
59 {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
61 dst: make([]complex64, 10),
62 x: []complex64{1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i},
63 y: []complex64{1, 1, 2, 1, 1, 1, 1, 2, 1, 1},
64 ex: []complex64{2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i}},
65 {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
67 dst: make([]complex64, 10),
68 x: []complex64{1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i},
69 y: []complex64{1, 1, 2, 1, 1, 1, 1, 2, 1, 1},
70 ex: []complex64{2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i}},
71 {incX: -2, incY: -2, incDst: -3, ix: 18, iy: 18, idst: 27,
73 dst: make([]complex64, 10),
74 x: []complex64{1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i},
75 y: []complex64{1, 1, 2, 1, 1, 1, 1, 2, 1, 1},
76 ex: []complex64{2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i}},
77 {incX: -2, incY: 2, incDst: -3, ix: 18, iy: 0, idst: 27,
79 dst: make([]complex64, 10),
80 x: []complex64{1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i},
81 y: []complex64{1, 1, 2, 1, 1, 1, 1, 2, 1, 1},
82 ex: []complex64{2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i}},
85 func TestAxpyUnitary(t *testing.T) {
86 var x_gd, y_gd complex64 = 1, 1
87 for cas, test := range tests {
88 xg_ln, yg_ln := 4+cas%2, 4+cas%3
89 test.x, test.y = guardVector(test.x, x_gd, xg_ln), guardVector(test.y, y_gd, yg_ln)
90 x, y := test.x[xg_ln:len(test.x)-xg_ln], test.y[yg_ln:len(test.y)-yg_ln]
91 AxpyUnitary(test.a, x, y)
92 for i := range test.ex {
93 if y[i] != test.ex[i] {
94 t.Errorf("Test %d Unexpected result at %d Got: %v Expected: %v", cas, i, y[i], test.ex[i])
97 if !isValidGuard(test.x, x_gd, xg_ln) {
98 t.Errorf("Test %d Guard violated in x vector %v %v", cas, test.x[:xg_ln], test.x[len(test.x)-xg_ln:])
100 if !isValidGuard(test.y, y_gd, yg_ln) {
101 t.Errorf("Test %d Guard violated in y vector %v %v", cas, test.y[:yg_ln], test.y[len(test.y)-yg_ln:])
106 func TestAxpyUnitaryTo(t *testing.T) {
107 var x_gd, y_gd, dst_gd complex64 = 1, 1, 0
108 for cas, test := range tests {
109 xg_ln, yg_ln := 4+cas%2, 4+cas%3
110 test.x, test.y = guardVector(test.x, x_gd, xg_ln), guardVector(test.y, y_gd, yg_ln)
111 test.dst = guardVector(test.dst, dst_gd, xg_ln)
112 x, y := test.x[xg_ln:len(test.x)-xg_ln], test.y[yg_ln:len(test.y)-yg_ln]
113 dst := test.dst[xg_ln : len(test.dst)-xg_ln]
114 AxpyUnitaryTo(dst, test.a, x, y)
115 for i := range test.ex {
116 if dst[i] != test.ex[i] {
117 t.Errorf("Test %d Unexpected result at %d Got: %v Expected: %v", cas, i, dst[i], test.ex[i])
120 if !isValidGuard(test.x, x_gd, xg_ln) {
121 t.Errorf("Test %d Guard violated in x vector %v %v", cas, test.x[:xg_ln], test.x[len(test.x)-xg_ln:])
123 if !isValidGuard(test.y, y_gd, yg_ln) {
124 t.Errorf("Test %d Guard violated in y vector %v %v", cas, test.y[:yg_ln], test.y[len(test.y)-yg_ln:])
126 if !isValidGuard(test.dst, dst_gd, xg_ln) {
127 t.Errorf("Test %d Guard violated in dst vector %v %v", cas, test.dst[:xg_ln], test.dst[len(test.dst)-xg_ln:])
132 func TestAxpyInc(t *testing.T) {
133 var x_gd, y_gd complex64 = 1, 1
134 for cas, test := range tests {
135 xg_ln, yg_ln := 4+cas%2, 4+cas%3
136 test.x, test.y = guardIncVector(test.x, x_gd, test.incX, xg_ln), guardIncVector(test.y, y_gd, test.incY, yg_ln)
137 x, y := test.x[xg_ln:len(test.x)-xg_ln], test.y[yg_ln:len(test.y)-yg_ln]
138 AxpyInc(test.a, x, y, uintptr(len(test.ex)), uintptr(test.incX), uintptr(test.incY), test.ix, test.iy)
139 for i := range test.ex {
140 if y[int(test.iy)+i*int(test.incY)] != test.ex[i] {
141 t.Errorf("Test %d Unexpected result at %d Got: %v Expected: %v", cas, i, y[i*int(test.incY)], test.ex[i])
144 checkValidIncGuard(t, test.x, x_gd, test.incX, xg_ln)
145 checkValidIncGuard(t, test.y, y_gd, test.incY, yg_ln)
149 func TestAxpyIncTo(t *testing.T) {
150 var x_gd, y_gd, dst_gd complex64 = 1, 1, 0
151 for cas, test := range tests {
152 xg_ln, yg_ln := 4+cas%2, 4+cas%3
153 test.x, test.y = guardIncVector(test.x, x_gd, test.incX, xg_ln), guardIncVector(test.y, y_gd, test.incY, yg_ln)
154 test.dst = guardIncVector(test.dst, dst_gd, test.incDst, xg_ln)
155 x, y := test.x[xg_ln:len(test.x)-xg_ln], test.y[yg_ln:len(test.y)-yg_ln]
156 dst := test.dst[xg_ln : len(test.dst)-xg_ln]
157 AxpyIncTo(dst, uintptr(test.incDst), test.idst, test.a, x, y, uintptr(len(test.ex)), uintptr(test.incX), uintptr(test.incY), test.ix, test.iy)
158 for i := range test.ex {
159 if dst[int(test.idst)+i*int(test.incDst)] != test.ex[i] {
160 t.Errorf("Test %d Unexpected result at %d Got: %v Expected: %v", cas, i, dst[i*int(test.incDst)], test.ex[i])
163 checkValidIncGuard(t, test.x, x_gd, test.incX, xg_ln)
164 checkValidIncGuard(t, test.y, y_gd, test.incY, yg_ln)
165 checkValidIncGuard(t, test.dst, dst_gd, test.incDst, xg_ln)