51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
|
package road
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/cyrilix/robocar-protobuf/go/events"
|
||
|
|
"gocv.io/x/gocv"
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDetector_Detect(t *testing.T) {
|
||
|
|
type fields struct {
|
||
|
|
}
|
||
|
|
type args struct {
|
||
|
|
imgName string
|
||
|
|
}
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
fields fields
|
||
|
|
args args
|
||
|
|
want []*events.Point
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "Straight ahead",
|
||
|
|
fields: fields{},
|
||
|
|
args: args{imgName: "testdata/input.jpg"},
|
||
|
|
want: []*events.Point{{X: 2, Y: 53}, {X: 48, Y: 37}, {X: 54, Y: 27}, {X: 72, Y: 26}, {X: 156, Y: 55}, {X: 159, Y: 118}, {X: 20, Y: 114}, {X: 3, Y: 119}},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, tt := range tests {
|
||
|
|
t.Run(tt.name, func(t *testing.T) {
|
||
|
|
d := NewDetector()
|
||
|
|
defer d.Close()
|
||
|
|
img := gocv.IMRead(tt.args.imgName, gocv.IMReadColor)
|
||
|
|
defer func(img *gocv.Mat) {
|
||
|
|
err := img.Close()
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("unable to close image: %v", err)
|
||
|
|
}
|
||
|
|
}(&img)
|
||
|
|
if img.Empty() {
|
||
|
|
t.Errorf("unable to open image %v", tt.args.imgName)
|
||
|
|
t.Fail()
|
||
|
|
}
|
||
|
|
|
||
|
|
if got, _ := d.Detect(&img); !reflect.DeepEqual(got, tt.want) {
|
||
|
|
t.Errorf("Detect() = %v, want %v", got, tt.want)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|