mirror of
https://github.com/gonum/gonum.git
synced 2025-10-05 15:16:59 +08:00
mat: Add Zero method to reset values of matrices (#819)
* mat: Add Zero method to reset values of matrices
This commit is contained in:
@@ -181,6 +181,38 @@ func TestVecDenseAtSet(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestVecDenseZero(t *testing.T) {
|
||||
// Elements that equal 1 should be set to zero, elements that equal -1
|
||||
// should remain unchanged.
|
||||
for _, test := range []*VecDense{
|
||||
&VecDense{
|
||||
mat: blas64.Vector{
|
||||
N: 5,
|
||||
Inc: 2,
|
||||
Data: []float64{
|
||||
1, -1,
|
||||
1, -1,
|
||||
1, -1,
|
||||
1, -1,
|
||||
1,
|
||||
},
|
||||
},
|
||||
},
|
||||
} {
|
||||
dataCopy := make([]float64, len(test.mat.Data))
|
||||
copy(dataCopy, test.mat.Data)
|
||||
test.Zero()
|
||||
for i, v := range test.mat.Data {
|
||||
if dataCopy[i] != -1 && v != 0 {
|
||||
t.Errorf("Matrix not zeroed in bounds")
|
||||
}
|
||||
if dataCopy[i] == -1 && v != -1 {
|
||||
t.Errorf("Matrix zeroed out of bounds")
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestVecDenseMul(t *testing.T) {
|
||||
method := func(receiver, a, b Matrix) {
|
||||
type mulVecer interface {
|
||||
|
Reference in New Issue
Block a user