24 lines
536 B
Go
24 lines
536 B
Go
package gocv
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/gonum/matrix/mat64"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGcvThreshold(t *testing.T) {
|
|
rvec := mat64.NewDense(3, 1, []float64{
|
|
3,
|
|
-0.3,
|
|
0.2,
|
|
})
|
|
rmat, _ := GcvThreshold(rvec)
|
|
|
|
assert.InDeltaSlice(t, []float64{0.59922526, 0.57799222, -0.55394411},
|
|
rmat.Row(nil, 0), 1e-5)
|
|
assert.InDeltaSlice(t, []float64{0.20413818, 0.558743, 0.80382452},
|
|
rmat.Row(nil, 1), 1e-5)
|
|
assert.InDeltaSlice(t, []float64{0.77411672, -0.5947531, 0.21682264},
|
|
rmat.Row(nil, 2), 1e-5)
|
|
}
|