| // Copyright ©2017 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 testblas |
| |
| import ( |
| "testing" |
| ) |
| |
| type Zgeruer interface { |
| Zgeru(m, n int, alpha complex128, x []complex128, incX int, y []complex128, incY int, a []complex128, lda int) |
| } |
| |
| func ZgeruTest(t *testing.T, impl Zgeruer) { |
| for tc, test := range []struct { |
| alpha complex128 |
| x []complex128 |
| incX int |
| y []complex128 |
| incY int |
| a []complex128 |
| |
| want []complex128 |
| }{ |
| { |
| incX: 1, |
| incY: 1, |
| alpha: 1 + 2i, |
| }, |
| { |
| incX: 1, |
| incY: 1, |
| alpha: 1 + 2i, |
| x: []complex128{1 + 13i, 18 + 15i, 10 + 18i}, |
| y: []complex128{15 + 12i, 4 + 8i, 5 + 16i, 19 + 12i}, |
| a: []complex128{ |
| 4 + 7i, 4 + 7i, 12 + 3i, 9 + 10i, |
| 3 + 3i, 1 + 2i, 17 + 17i, 9 + 18i, |
| 14 + 12i, 9 + 16i, 1 + 1i, 9 + 1i, |
| }, |
| want: []complex128{ |
| -551 - 68i, -216 - 133i, -353 - 322i, -646 - 5i, |
| -789 + 624i, -455 + 110i, -859 + 80i, -831 + 843i, |
| -832 + 270i, -399 - 40i, -737 - 225i, -941 + 411i, |
| }, |
| }, |
| { |
| incX: 7, |
| incY: 13, |
| alpha: 1 + 2i, |
| x: []complex128{1 + 13i, 18 + 15i, 10 + 18i}, |
| y: []complex128{15 + 12i, 4 + 8i, 5 + 16i, 19 + 12i}, |
| a: []complex128{ |
| 4 + 7i, 4 + 7i, 12 + 3i, 9 + 10i, |
| 3 + 3i, 1 + 2i, 17 + 17i, 9 + 18i, |
| 14 + 12i, 9 + 16i, 1 + 1i, 9 + 1i, |
| }, |
| want: []complex128{ |
| -551 - 68i, -216 - 133i, -353 - 322i, -646 - 5i, |
| -789 + 624i, -455 + 110i, -859 + 80i, -831 + 843i, |
| -832 + 270i, -399 - 40i, -737 - 225i, -941 + 411i, |
| }, |
| }, |
| { |
| incX: 1, |
| incY: 13, |
| alpha: 1 + 2i, |
| x: []complex128{1 + 13i, 18 + 15i, 10 + 18i}, |
| y: []complex128{15 + 12i, 4 + 8i, 5 + 16i, 19 + 12i}, |
| a: []complex128{ |
| 4 + 7i, 4 + 7i, 12 + 3i, 9 + 10i, |
| 3 + 3i, 1 + 2i, 17 + 17i, 9 + 18i, |
| 14 + 12i, 9 + 16i, 1 + 1i, 9 + 1i, |
| }, |
| want: []complex128{ |
| -551 - 68i, -216 - 133i, -353 - 322i, -646 - 5i, |
| -789 + 624i, -455 + 110i, -859 + 80i, -831 + 843i, |
| -832 + 270i, -399 - 40i, -737 - 225i, -941 + 411i, |
| }, |
| }, |
| { |
| incX: 1, |
| incY: -13, |
| alpha: 1 + 2i, |
| x: []complex128{1 + 13i, 18 + 15i, 10 + 18i}, |
| y: []complex128{19 + 12i, 5 + 16i, 4 + 8i, 15 + 12i}, |
| a: []complex128{ |
| 4 + 7i, 4 + 7i, 12 + 3i, 9 + 10i, |
| 3 + 3i, 1 + 2i, 17 + 17i, 9 + 18i, |
| 14 + 12i, 9 + 16i, 1 + 1i, 9 + 1i, |
| }, |
| want: []complex128{ |
| -551 - 68i, -216 - 133i, -353 - 322i, -646 - 5i, |
| -789 + 624i, -455 + 110i, -859 + 80i, -831 + 843i, |
| -832 + 270i, -399 - 40i, -737 - 225i, -941 + 411i, |
| }, |
| }, |
| { |
| incX: 7, |
| incY: 1, |
| alpha: 1 + 2i, |
| x: []complex128{1 + 13i, 18 + 15i, 10 + 18i}, |
| y: []complex128{15 + 12i, 4 + 8i, 5 + 16i, 19 + 12i}, |
| a: []complex128{ |
| 4 + 7i, 4 + 7i, 12 + 3i, 9 + 10i, |
| 3 + 3i, 1 + 2i, 17 + 17i, 9 + 18i, |
| 14 + 12i, 9 + 16i, 1 + 1i, 9 + 1i, |
| }, |
| want: []complex128{ |
| -551 - 68i, -216 - 133i, -353 - 322i, -646 - 5i, |
| -789 + 624i, -455 + 110i, -859 + 80i, -831 + 843i, |
| -832 + 270i, -399 - 40i, -737 - 225i, -941 + 411i, |
| }, |
| }, |
| { |
| incX: -7, |
| incY: 1, |
| alpha: 1 + 2i, |
| x: []complex128{10 + 18i, 18 + 15i, 1 + 13i}, |
| y: []complex128{15 + 12i, 4 + 8i, 5 + 16i, 19 + 12i}, |
| a: []complex128{ |
| 4 + 7i, 4 + 7i, 12 + 3i, 9 + 10i, |
| 3 + 3i, 1 + 2i, 17 + 17i, 9 + 18i, |
| 14 + 12i, 9 + 16i, 1 + 1i, 9 + 1i, |
| }, |
| want: []complex128{ |
| -551 - 68i, -216 - 133i, -353 - 322i, -646 - 5i, |
| -789 + 624i, -455 + 110i, -859 + 80i, -831 + 843i, |
| -832 + 270i, -399 - 40i, -737 - 225i, -941 + 411i, |
| }, |
| }, |
| { |
| incX: -7, |
| incY: -13, |
| alpha: 1 + 2i, |
| x: []complex128{10 + 18i, 18 + 15i, 1 + 13i}, |
| y: []complex128{19 + 12i, 5 + 16i, 4 + 8i, 15 + 12i}, |
| a: []complex128{ |
| 4 + 7i, 4 + 7i, 12 + 3i, 9 + 10i, |
| 3 + 3i, 1 + 2i, 17 + 17i, 9 + 18i, |
| 14 + 12i, 9 + 16i, 1 + 1i, 9 + 1i, |
| }, |
| want: []complex128{ |
| -551 - 68i, -216 - 133i, -353 - 322i, -646 - 5i, |
| -789 + 624i, -455 + 110i, -859 + 80i, -831 + 843i, |
| -832 + 270i, -399 - 40i, -737 - 225i, -941 + 411i, |
| }, |
| }, |
| { |
| incX: 1, |
| incY: 1, |
| alpha: 1 + 2i, |
| x: []complex128{5 + 16i, 12 + 19i, 9 + 7i, 2 + 4i}, |
| y: []complex128{18 + 7i, 20 + 15i, 12 + 14i}, |
| a: []complex128{ |
| 8 + 17i, 2 + 2i, 8 + 17i, |
| 1 + 10i, 10 + 15i, 4 + 18i, |
| 11 + 3i, 15 + 7i, 12 + 15i, |
| 20 + 10i, 8 + 13i, 19 + 10i, |
| }, |
| want: []complex128{ |
| -660 + 296i, -928 + 117i, -680 - 49i, |
| -768 + 602i, -1155 + 485i, -910 + 170i, |
| -254 + 418i, -460 + 432i, -398 + 245i, |
| -144 + 112i, -232 + 83i, -165 + 22i, |
| }, |
| }, |
| { |
| incX: 7, |
| incY: 13, |
| alpha: 1 + 2i, |
| x: []complex128{5 + 16i, 12 + 19i, 9 + 7i, 2 + 4i}, |
| y: []complex128{18 + 7i, 20 + 15i, 12 + 14i}, |
| a: []complex128{ |
| 8 + 17i, 2 + 2i, 8 + 17i, |
| 1 + 10i, 10 + 15i, 4 + 18i, |
| 11 + 3i, 15 + 7i, 12 + 15i, |
| 20 + 10i, 8 + 13i, 19 + 10i, |
| }, |
| want: []complex128{ |
| -660 + 296i, -928 + 117i, -680 - 49i, |
| -768 + 602i, -1155 + 485i, -910 + 170i, |
| -254 + 418i, -460 + 432i, -398 + 245i, |
| -144 + 112i, -232 + 83i, -165 + 22i, |
| }, |
| }, |
| { |
| incX: -7, |
| incY: -13, |
| alpha: 1 + 2i, |
| x: []complex128{2 + 4i, 9 + 7i, 12 + 19i, 5 + 16i}, |
| y: []complex128{12 + 14i, 20 + 15i, 18 + 7i}, |
| a: []complex128{ |
| 8 + 17i, 2 + 2i, 8 + 17i, |
| 1 + 10i, 10 + 15i, 4 + 18i, |
| 11 + 3i, 15 + 7i, 12 + 15i, |
| 20 + 10i, 8 + 13i, 19 + 10i, |
| }, |
| want: []complex128{ |
| -660 + 296i, -928 + 117i, -680 - 49i, |
| -768 + 602i, -1155 + 485i, -910 + 170i, |
| -254 + 418i, -460 + 432i, -398 + 245i, |
| -144 + 112i, -232 + 83i, -165 + 22i, |
| }, |
| }, |
| { |
| incX: -7, |
| incY: -13, |
| alpha: 0, |
| x: []complex128{5 + 16i, 12 + 19i, 9 + 7i, 2 + 4i}, |
| y: []complex128{18 + 7i, 20 + 15i, 12 + 14i}, |
| a: []complex128{ |
| 8 + 17i, 2 + 2i, 8 + 17i, |
| 1 + 10i, 10 + 15i, 4 + 18i, |
| 11 + 3i, 15 + 7i, 12 + 15i, |
| 20 + 10i, 8 + 13i, 19 + 10i, |
| }, |
| want: []complex128{ |
| 8 + 17i, 2 + 2i, 8 + 17i, |
| 1 + 10i, 10 + 15i, 4 + 18i, |
| 11 + 3i, 15 + 7i, 12 + 15i, |
| 20 + 10i, 8 + 13i, 19 + 10i, |
| }, |
| }, |
| } { |
| m := len(test.x) |
| n := len(test.y) |
| incX := test.incX |
| incY := test.incY |
| |
| for _, lda := range []int{max(1, n), n + 20} { |
| x := makeZVector(test.x, incX) |
| xCopy := make([]complex128, len(x)) |
| copy(xCopy, x) |
| |
| y := makeZVector(test.y, incY) |
| yCopy := make([]complex128, len(y)) |
| copy(yCopy, y) |
| |
| a := makeZGeneral(test.a, m, n, lda) |
| want := makeZGeneral(test.want, m, n, lda) |
| |
| impl.Zgeru(m, n, test.alpha, x, incX, y, incY, a, lda) |
| |
| if !zsame(x, xCopy) { |
| t.Errorf("Case %v: unexpected modification of x", tc) |
| } |
| if !zsame(y, yCopy) { |
| t.Errorf("Case %v: unexpected modification of y", tc) |
| } |
| if !zsame(want, a) { |
| t.Errorf("Case %v: unexpected result\nwant %v\ngot %v", tc, want, a) |
| } |
| } |
| } |
| } |