func _TestBKpivot1U(t *testing.T) { Ldata := [][]float64{ []float64{7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0}, []float64{1.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0}, []float64{1.0, 2.0, 5.0, 4.0, 3.0, 5.0, 1.0}, []float64{1.0, 2.0, 3.0, 4.0, 3.0, 2.0, 1.0}, []float64{1.0, 5.0, 3.0, 4.0, 3.0, 2.0, 1.0}, []float64{1.0, 2.0, 3.0, 4.0, 5.0, 2.0, 1.0}, []float64{1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 1.0}} Bdata := [][]float64{ []float64{10.0, 20.0}, []float64{10.0, 20.0}, []float64{10.0, 20.0}, []float64{10.0, 20.0}, []float64{10.0, 20.0}, []float64{10.0, 20.0}, []float64{10.0, 20.0}} A := matrix.FloatMatrixFromTable(Ldata, matrix.RowOrder) X := matrix.FloatMatrixFromTable(Bdata, matrix.RowOrder) N := A.Rows() B := matrix.FloatZeros(N, 2) MultSym(B, A, X, 1.0, 0.0, UPPER|LEFT) //t.Logf("initial B:\n%v\n", B) //N := 8 //A := matrix.FloatUniformSymmetric(N) W := matrix.FloatWithValue(A.Rows(), 5, 0.0) nb := 0 ipiv := make([]int, N, N) L, _ := DecomposeBK(A.Copy(), W, ipiv, UPPER, nb) t.Logf("ipiv: %v\n", ipiv) t.Logf("L:\n%v\n", L) nb = 4 //ipiv0 := make([]int, N, N) //L0, _ := DecomposeBK(A.Copy(), W, ipiv0, LOWER, nb) //t.Logf("ipiv: %v\n", ipiv0) //t.Logf("L:\n%v\n", L0) //B0 := B.Copy() //SolveBK(B0, L0, ipiv0, LOWER) //t.Logf("B0:\n%v\n", B0) ipiv2 := make([]int32, N, N) lapack.SytrfFloat(A, ipiv2, linalg.OptUpper) t.Logf("ipiv2: %v\n", ipiv2) t.Logf("lapack A:\n%v\n", A) //lapack.Sytrs(A, B, ipiv2, linalg.OptLower) //t.Logf("lapack B:\n%v\n", B) //t.Logf("B == B0: %v\n", B.AllClose(B0)) }
func ldlbkDecomposeUpperTest(A *matrix.FloatMatrix, t *testing.T) { TriU(A) N := A.Cols() nb := 0 W := matrix.FloatZeros(A.Rows(), 5) ipiv := make([]int, N, N) L, _ := DecomposeBK(A.Copy(), W, ipiv, UPPER, nb) t.Logf("L:\n%v\n", L) t.Logf("unblked ipiv: %v\n", ipiv) ipiv0 := make([]int, N, N) nb = 4 L0, _ := DecomposeBK(A.Copy(), W, ipiv0, UPPER, nb) t.Logf("L:\n%v\n", L0) t.Logf("blked ipiv: %v\n", ipiv0) ipiv2 := make([]int32, N, N) lapack.SytrfFloat(A, ipiv2, linalg.OptUpper) t.Logf("lapack A:\n%v\n", A) t.Logf("lapack ipiv: %v\n", ipiv2) }