| // Copyright ©2015 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 f32_test |
| |
| import ( |
| "testing" |
| |
| . "gonum.org/v1/gonum/internal/asm/f32" |
| ) |
| |
| var tests = []struct { |
| incX, incY, incDst uintptr |
| ix, iy, idst uintptr |
| a float32 |
| dst, x, y []float32 |
| ex []float32 |
| }{ |
| {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0, |
| a: 3, |
| dst: []float32{5}, |
| x: []float32{2}, |
| y: []float32{1}, |
| ex: []float32{7}}, |
| {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0, |
| a: 5, |
| dst: []float32{0, 0, 0}, |
| x: []float32{0, 0, 0}, |
| y: []float32{1, 1, 1}, |
| ex: []float32{1, 1, 1}}, |
| {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0, |
| a: 5, |
| dst: []float32{0, 0, 0}, |
| x: []float32{0, 0}, |
| y: []float32{1, 1, 1}, |
| ex: []float32{1, 1}}, |
| {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0, |
| a: -1, |
| dst: []float32{-1, -1, -1}, |
| x: []float32{1, 1, 1}, |
| y: []float32{1, 2, 1}, |
| ex: []float32{0, 1, 0}}, |
| {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0, |
| a: -1, |
| dst: []float32{1, 1, 1}, |
| x: []float32{1, 2, 1}, |
| y: []float32{-1, -2, -1}, |
| ex: []float32{-2, -4, -2}}, |
| {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0, |
| a: 2.5, |
| dst: []float32{1, 1, 1, 1, 1}, |
| x: []float32{1, 2, 3, 2, 1}, |
| y: []float32{0, 0, 0, 0, 0}, |
| ex: []float32{2.5, 5, 7.5, 5, 2.5}}, |
| {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0, // Run big test twice, once aligned once unaligned. |
| a: 16.5, |
| dst: make([]float32, 20), |
| x: []float32{.5, .5, .5, .5, .5, .5, .5, .5, .5, .5, .5, .5, .5, .5, .5, .5, .5, .5, .5, .5}, |
| y: []float32{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10}, |
| ex: []float32{9.25, 10.25, 11.25, 12.25, 13.25, 14.25, 15.25, 16.25, 17.25, 18.25, 9.25, 10.25, 11.25, 12.25, 13.25, 14.25, 15.25, 16.25, 17.25, 18.25}}, |
| {incX: 2, incY: 2, incDst: 3, ix: 0, iy: 0, idst: 0, |
| a: 16.5, |
| dst: make([]float32, 10), |
| x: []float32{.5, .5, .5, .5, .5, .5, .5, .5, .5, .5}, |
| y: []float32{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}, |
| ex: []float32{9.25, 10.25, 11.25, 12.25, 13.25, 14.25, 15.25, 16.25, 17.25, 18.25}}, |
| } |
| |
| func TestAxpyUnitary(t *testing.T) { |
| for j, v := range tests { |
| gdLn := 4 + j%2 |
| v.x, v.y = guardVector(v.x, 1, gdLn), guardVector(v.y, 1, gdLn) |
| x, y := v.x[gdLn:len(v.x)-gdLn], v.y[gdLn:len(v.y)-gdLn] |
| AxpyUnitary(v.a, x, y) |
| for i := range v.ex { |
| if !same(y[i], v.ex[i]) { |
| t.Error("Test", j, "Unexpected result at", i, "Got:", int(y[i]), "Expected:", v.ex[i]) |
| } |
| } |
| if !isValidGuard(v.x, 1, gdLn) { |
| t.Error("Test", j, "Guard violated in x vector", v.x[:gdLn], v.x[len(v.x)-gdLn:]) |
| } |
| if !isValidGuard(v.y, 1, gdLn) { |
| t.Error("Test", j, "Guard violated in y vector", v.y[:gdLn], v.y[len(v.x)-gdLn:]) |
| } |
| } |
| } |
| |
| func TestAxpyUnitaryTo(t *testing.T) { |
| for j, v := range tests { |
| gdLn := 4 + j%2 |
| v.x, v.y = guardVector(v.x, 1, gdLn), guardVector(v.y, 1, gdLn) |
| v.dst = guardVector(v.dst, 0, gdLn) |
| x, y := v.x[gdLn:len(v.x)-gdLn], v.y[gdLn:len(v.y)-gdLn] |
| dst := v.dst[gdLn : len(v.dst)-gdLn] |
| AxpyUnitaryTo(dst, v.a, x, y) |
| for i := range v.ex { |
| if !same(v.ex[i], dst[i]) { |
| t.Error("Test", j, "Unexpected result at", i, "Got:", dst[i], "Expected:", v.ex[i]) |
| } |
| } |
| if !isValidGuard(v.x, 1, gdLn) { |
| t.Error("Test", j, "Guard violated in x vector", v.x[:gdLn], v.x[len(v.x)-gdLn:]) |
| } |
| if !isValidGuard(v.y, 1, gdLn) { |
| t.Error("Test", j, "Guard violated in y vector", v.y[:gdLn], v.y[len(v.x)-gdLn:]) |
| } |
| if !isValidGuard(v.dst, 0, gdLn) { |
| t.Error("Test", j, "Guard violated in x vector", v.x[:gdLn], v.x[len(v.x)-gdLn:]) |
| } |
| } |
| } |
| |
| func TestAxpyInc(t *testing.T) { |
| for j, v := range tests { |
| gdLn := 4 + j%2 |
| v.x, v.y = guardIncVector(v.x, 1, int(v.incX), gdLn), guardIncVector(v.y, 1, int(v.incY), gdLn) |
| x, y := v.x[gdLn:len(v.x)-gdLn], v.y[gdLn:len(v.y)-gdLn] |
| AxpyInc(v.a, x, y, uintptr(len(v.ex)), v.incX, v.incY, v.ix, v.iy) |
| for i := range v.ex { |
| if !same(y[i*int(v.incY)], v.ex[i]) { |
| t.Error("Test", j, "Unexpected result at", i, "Got:", y[i*int(v.incY)], "Expected:", v.ex[i]) |
| t.Error("Result:", y) |
| t.Error("Expect:", v.ex) |
| } |
| } |
| checkValidIncGuard(t, v.x, 1, int(v.incX), gdLn) |
| checkValidIncGuard(t, v.y, 1, int(v.incY), gdLn) |
| } |
| } |
| |
| func TestAxpyIncTo(t *testing.T) { |
| for j, v := range tests { |
| gdLn := 4 + j%2 |
| v.x, v.y = guardIncVector(v.x, 1, int(v.incX), gdLn), guardIncVector(v.y, 1, int(v.incY), gdLn) |
| v.dst = guardIncVector(v.dst, 0, int(v.incDst), gdLn) |
| x, y := v.x[gdLn:len(v.x)-gdLn], v.y[gdLn:len(v.y)-gdLn] |
| dst := v.dst[gdLn : len(v.dst)-gdLn] |
| AxpyIncTo(dst, v.incDst, v.idst, v.a, x, y, uintptr(len(v.ex)), v.incX, v.incY, v.ix, v.iy) |
| for i := range v.ex { |
| if !same(dst[i*int(v.incDst)], v.ex[i]) { |
| t.Error("Test", j, "Unexpected result at", i, "Got:", dst[i*int(v.incDst)], "Expected:", v.ex[i]) |
| t.Error(v.dst) |
| t.Error(v.ex) |
| } |
| } |
| checkValidIncGuard(t, v.x, 1, int(v.incX), gdLn) |
| checkValidIncGuard(t, v.y, 1, int(v.incY), gdLn) |
| checkValidIncGuard(t, v.dst, 0, int(v.incDst), gdLn) |
| } |
| } |
| |
| func TestSum(t *testing.T) { |
| var srcGd float32 = -1 |
| for j, v := range []struct { |
| src []float32 |
| expect float32 |
| }{ |
| { |
| src: []float32{}, |
| expect: 0, |
| }, |
| { |
| src: []float32{1}, |
| expect: 1, |
| }, |
| { |
| src: []float32{nan}, |
| expect: nan, |
| }, |
| { |
| src: []float32{1, 2, 3}, |
| expect: 6, |
| }, |
| { |
| src: []float32{1, -4, 3}, |
| expect: 0, |
| }, |
| { |
| src: []float32{1, 2, 3, 4}, |
| expect: 10, |
| }, |
| { |
| src: []float32{1, 1, nan, 1, 1}, |
| expect: nan, |
| }, |
| { |
| src: []float32{inf, 4, nan, -inf, 9}, |
| expect: nan, |
| }, |
| { |
| src: []float32{1, 1, 1, 1, 9, 1, 1, 1, 2, 1, 1, 1, 1, 1, 5, 1}, |
| expect: 29, |
| }, |
| { |
| src: []float32{1, 1, 1, 1, 9, 1, 1, 1, 2, 1, 1, 1, 1, 1, 5, 11, 1, 1, 1, 9, 1, 1, 1, 2, 1, 1, 1, 1, 1, 5, 1}, |
| expect: 67, |
| }, |
| } { |
| for _, i := range [4]int{0, 1, 2, 3} { |
| gdLn := 4 + j%4 + i |
| gsrc := guardVector(v.src, srcGd, gdLn) |
| src := gsrc[gdLn : len(gsrc)-gdLn] |
| ret := Sum(src) |
| if !same(ret, v.expect) { |
| t.Errorf("Test %d Sum error Got: %v Expected: %v", j, ret, v.expect) |
| } |
| if !isValidGuard(gsrc, srcGd, gdLn) { |
| t.Errorf("Test %d Guard violated in src vector %v %v", j, gsrc[:gdLn], gsrc[len(gsrc)-gdLn:]) |
| } |
| } |
| } |
| } |