mirror of
https://github.com/sjwhitworth/golearn.git
synced 2025-04-26 13:49:14 +08:00
37 lines
803 B
Go
37 lines
803 B
Go
package pairwise
|
|
|
|
import (
|
|
"testing"
|
|
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
"gonum.org/v1/gonum/mat"
|
|
)
|
|
|
|
func TestPolyKernel(t *testing.T) {
|
|
var vectorX, vectorY *mat.Dense
|
|
polyKernel := NewPolyKernel(3)
|
|
|
|
Convey("Given two vectors", t, func() {
|
|
vectorX = mat.NewDense(3, 1, []float64{1, 2, 3})
|
|
vectorY = mat.NewDense(3, 1, []float64{2, 4, 5})
|
|
|
|
Convey("When doing inner product", func() {
|
|
result := polyKernel.InnerProduct(vectorX, vectorY)
|
|
|
|
Convey("The result should be 17576", func() {
|
|
So(result, ShouldEqual, 17576)
|
|
})
|
|
})
|
|
|
|
Convey("When calculating distance", func() {
|
|
result := polyKernel.Distance(vectorX, vectorY)
|
|
|
|
Convey("The result should alomost equal 31.622776601683793", func() {
|
|
So(result, ShouldAlmostEqual, 31.622776601683793)
|
|
})
|
|
|
|
})
|
|
|
|
})
|
|
}
|