// 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)
			}
		}
	}
}