mirror of
https://github.com/gonum/gonum.git
synced 2025-10-05 23:26:52 +08:00
232 lines
8.3 KiB
Go
232 lines
8.3 KiB
Go
// Copyright ©2016 The gonum Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package c128
|
|
|
|
import "testing"
|
|
|
|
var tests = []struct {
|
|
incX, incY, incDst int
|
|
ix, iy, idst uintptr
|
|
a complex128
|
|
dst, x, y []complex128
|
|
ex []complex128
|
|
}{
|
|
{incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
|
|
a: 1 + 1i,
|
|
dst: []complex128{5},
|
|
x: []complex128{1},
|
|
y: []complex128{1i},
|
|
ex: []complex128{1 + 2i}},
|
|
{incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
|
|
a: 1 + 2i,
|
|
dst: []complex128{0, 0, 0},
|
|
x: []complex128{0, 0, 0},
|
|
y: []complex128{1, 1, 1},
|
|
ex: []complex128{1, 1, 1}},
|
|
{incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
|
|
a: 1 + 2i,
|
|
dst: []complex128{0, 0, 0},
|
|
x: []complex128{0, 0},
|
|
y: []complex128{1, 1, 1},
|
|
ex: []complex128{1, 1}},
|
|
{incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
|
|
a: 1 + 2i,
|
|
dst: []complex128{1i, 1i, 1i},
|
|
x: []complex128{1i, 1i, 1i},
|
|
y: []complex128{1, 2, 1},
|
|
ex: []complex128{-1 + 1i, 1i, -1 + 1i}},
|
|
{incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
|
|
a: -1i,
|
|
dst: []complex128{1i, 1i, 1i},
|
|
x: []complex128{1i, 1i, 1i},
|
|
y: []complex128{1, 2, 1},
|
|
ex: []complex128{2, 3, 2}},
|
|
{incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
|
|
a: -1i,
|
|
dst: []complex128{1i, 1i, 1i},
|
|
x: []complex128{1i, 1i, 1i, 1i, 1i}[1:4],
|
|
y: []complex128{1, 1, 2, 1, 1}[1:4],
|
|
ex: []complex128{2, 3, 2}},
|
|
{incX: 2, incY: 4, incDst: 3, ix: 0, iy: 0, idst: 0,
|
|
a: -2,
|
|
dst: []complex128{1i, 1i, 1i, 1i, 1i},
|
|
x: []complex128{2 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 2 + 1i},
|
|
y: []complex128{1, 1, 2, 1, 1},
|
|
ex: []complex128{-3 - 2i, -3 - 2i, -2 - 2i, -3 - 2i, -3 - 2i}},
|
|
// Run big test twice, once aligned once unaligned.
|
|
{incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
|
|
a: 1 - 1i,
|
|
dst: make([]complex128, 10),
|
|
x: []complex128{1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i},
|
|
y: []complex128{1, 1, 2, 1, 1, 1, 1, 2, 1, 1},
|
|
ex: []complex128{2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i}},
|
|
{incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0,
|
|
a: 1 - 1i,
|
|
dst: make([]complex128, 10),
|
|
x: []complex128{1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i},
|
|
y: []complex128{1, 1, 2, 1, 1, 1, 1, 2, 1, 1},
|
|
ex: []complex128{2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i}},
|
|
{incX: -2, incY: -2, incDst: -3, ix: 18, iy: 18, idst: 27,
|
|
a: 1 - 1i,
|
|
dst: make([]complex128, 10),
|
|
x: []complex128{1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i},
|
|
y: []complex128{1, 1, 2, 1, 1, 1, 1, 2, 1, 1},
|
|
ex: []complex128{2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i}},
|
|
{incX: -2, incY: 2, incDst: -3, ix: 18, iy: 0, idst: 27,
|
|
a: 1 - 1i,
|
|
dst: make([]complex128, 10),
|
|
x: []complex128{1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i, 1i},
|
|
y: []complex128{1, 1, 2, 1, 1, 1, 1, 2, 1, 1},
|
|
ex: []complex128{2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 2 + 1i, 3 + 1i, 2 + 1i, 2 + 1i}},
|
|
}
|
|
|
|
func guardVector(vec []complex128, guard_val complex128, guard_len int) (guarded []complex128) {
|
|
guarded = make([]complex128, len(vec)+guard_len*2)
|
|
copy(guarded[guard_len:], vec)
|
|
for i := 0; i < guard_len; i++ {
|
|
guarded[i] = guard_val
|
|
guarded[len(guarded)-1-i] = guard_val
|
|
}
|
|
return guarded
|
|
}
|
|
|
|
func isValidGuard(vec []complex128, guard_val complex128, guard_len int) bool {
|
|
for i := 0; i < guard_len; i++ {
|
|
if vec[i] != guard_val || vec[len(vec)-1-i] != guard_val {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func TestAxpyUnitary(t *testing.T) {
|
|
var x_gd, y_gd complex128 = 1, 1
|
|
for cas, test := range tests {
|
|
xg_ln, yg_ln := 4+cas%2, 4+cas%3
|
|
test.x, test.y = guardVector(test.x, x_gd, xg_ln), guardVector(test.y, y_gd, yg_ln)
|
|
x, y := test.x[xg_ln:len(test.x)-xg_ln], test.y[yg_ln:len(test.y)-yg_ln]
|
|
AxpyUnitary(test.a, x, y)
|
|
for i := range test.ex {
|
|
if y[i] != test.ex[i] {
|
|
t.Errorf("Test %d Unexpected result at %d Got: %v Expected: %v", cas, i, y[i], test.ex[i])
|
|
}
|
|
}
|
|
if !isValidGuard(test.x, x_gd, xg_ln) {
|
|
t.Errorf("Test %d Guard violated in x vector %v %v", cas, test.x[:xg_ln], test.x[len(test.x)-xg_ln:])
|
|
}
|
|
if !isValidGuard(test.y, y_gd, yg_ln) {
|
|
t.Errorf("Test %d Guard violated in y vector %v %v", cas, test.y[:yg_ln], test.y[len(test.y)-yg_ln:])
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAxpyUnitaryTo(t *testing.T) {
|
|
var x_gd, y_gd, dst_gd complex128 = 1, 1, 0
|
|
for cas, test := range tests {
|
|
xg_ln, yg_ln := 4+cas%2, 4+cas%3
|
|
test.x, test.y = guardVector(test.x, x_gd, xg_ln), guardVector(test.y, y_gd, yg_ln)
|
|
test.dst = guardVector(test.dst, dst_gd, xg_ln)
|
|
x, y := test.x[xg_ln:len(test.x)-xg_ln], test.y[yg_ln:len(test.y)-yg_ln]
|
|
dst := test.dst[xg_ln : len(test.dst)-xg_ln]
|
|
AxpyUnitaryTo(dst, test.a, x, y)
|
|
for i := range test.ex {
|
|
if dst[i] != test.ex[i] {
|
|
t.Errorf("Test %d Unexpected result at %d Got: %v Expected: %v", cas, i, dst[i], test.ex[i])
|
|
}
|
|
}
|
|
if !isValidGuard(test.x, x_gd, xg_ln) {
|
|
t.Errorf("Test %d Guard violated in x vector %v %v", cas, test.x[:xg_ln], test.x[len(test.x)-xg_ln:])
|
|
}
|
|
if !isValidGuard(test.y, y_gd, yg_ln) {
|
|
t.Errorf("Test %d Guard violated in y vector %v %v", cas, test.y[:yg_ln], test.y[len(test.y)-yg_ln:])
|
|
}
|
|
if !isValidGuard(test.dst, dst_gd, xg_ln) {
|
|
t.Errorf("Test %d Guard violated in dst vector %v %v", cas, test.dst[:xg_ln], test.dst[len(test.dst)-xg_ln:])
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
func guardIncVector(vec []complex128, guard_val complex128, incV uintptr, guard_len int) (guarded []complex128) {
|
|
inc := int(incV)
|
|
s_ln := len(vec) * inc
|
|
if inc < 0 {
|
|
s_ln = len(vec) * -inc
|
|
}
|
|
guarded = make([]complex128, s_ln+guard_len*2)
|
|
for i, cas := 0, 0; i < len(guarded); i++ {
|
|
switch {
|
|
case i < guard_len, i > guard_len+s_ln:
|
|
guarded[i] = guard_val
|
|
case (i-guard_len)%(inc) == 0 && cas < len(vec):
|
|
guarded[i] = vec[cas]
|
|
cas++
|
|
default:
|
|
guarded[i] = guard_val
|
|
}
|
|
}
|
|
return guarded
|
|
}
|
|
|
|
func checkValidIncGuard(t *testing.T, vec []complex128, guard_val complex128, incV uintptr, guard_len int) {
|
|
inc := int(incV)
|
|
s_ln := len(vec) - 2*guard_len
|
|
if inc < 0 {
|
|
s_ln = len(vec) * -inc
|
|
}
|
|
|
|
for i := range vec {
|
|
switch {
|
|
case vec[i] == guard_val:
|
|
// Correct value
|
|
case i < guard_len:
|
|
t.Errorf("Front guard violated at %d %v", i, vec[:guard_len])
|
|
case i > guard_len+s_ln:
|
|
t.Errorf("Back guard violated at %d %v", i-guard_len-s_ln, vec[guard_len+s_ln:])
|
|
case (i-guard_len)%inc == 0 && (i-guard_len)/inc < len(vec):
|
|
// Ignore input values
|
|
default:
|
|
t.Errorf("Internal guard violated at %d %v", i-guard_len, vec[guard_len:guard_len+s_ln])
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAxpyInc(t *testing.T) {
|
|
var x_gd, y_gd complex128 = 1, 1
|
|
for cas, test := range tests {
|
|
xg_ln, yg_ln := 4+cas%2, 4+cas%3
|
|
test.x, test.y = guardIncVector(test.x, x_gd, uintptr(test.incX), xg_ln), guardIncVector(test.y, y_gd, uintptr(test.incY), yg_ln)
|
|
x, y := test.x[xg_ln:len(test.x)-xg_ln], test.y[yg_ln:len(test.y)-yg_ln]
|
|
AxpyInc(test.a, x, y, uintptr(len(test.ex)), uintptr(test.incX), uintptr(test.incY), test.ix, test.iy)
|
|
for i := range test.ex {
|
|
if y[int(test.iy)+i*int(test.incY)] != test.ex[i] {
|
|
t.Errorf("Test %d Unexpected result at %d Got: %v Expected: %v", cas, i, y[i*int(test.incY)], test.ex[i])
|
|
}
|
|
}
|
|
checkValidIncGuard(t, test.x, x_gd, uintptr(test.incX), xg_ln)
|
|
checkValidIncGuard(t, test.y, y_gd, uintptr(test.incY), yg_ln)
|
|
}
|
|
}
|
|
|
|
func TestAxpyIncTo(t *testing.T) {
|
|
var x_gd, y_gd, dst_gd complex128 = 1, 1, 0
|
|
for cas, test := range tests {
|
|
xg_ln, yg_ln := 4+cas%2, 4+cas%3
|
|
test.x, test.y = guardIncVector(test.x, x_gd, uintptr(test.incX), xg_ln), guardIncVector(test.y, y_gd, uintptr(test.incY), yg_ln)
|
|
test.dst = guardIncVector(test.dst, dst_gd, uintptr(test.incDst), xg_ln)
|
|
x, y := test.x[xg_ln:len(test.x)-xg_ln], test.y[yg_ln:len(test.y)-yg_ln]
|
|
dst := test.dst[xg_ln : len(test.dst)-xg_ln]
|
|
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)
|
|
for i := range test.ex {
|
|
if dst[int(test.idst)+i*int(test.incDst)] != test.ex[i] {
|
|
t.Errorf("Test %d Unexpected result at %d Got: %v Expected: %v", cas, i, dst[i*int(test.incDst)], test.ex[i])
|
|
}
|
|
}
|
|
checkValidIncGuard(t, test.x, x_gd, uintptr(test.incX), xg_ln)
|
|
checkValidIncGuard(t, test.y, y_gd, uintptr(test.incY), yg_ln)
|
|
checkValidIncGuard(t, test.dst, dst_gd, uintptr(test.incDst), xg_ln)
|
|
}
|
|
}
|