robocar-steering-tflite-edgetpu/pkg/tools/tools_test.go

package tools

import (
	"testing"
)

func Test_LinearBin(t *testing.T) {
	type args struct {
		arr    []byte
		n      int
		offset int
		r      float64
	}
	tests := []struct {
		name  string
		args  args
		want  float64
		want1 float64
	}{
		{
			name: "default",
			args: args{
				arr:    []byte{0, 0, 0, 0, 0, 0, 0, 255, 0, 0, 0, 0, 0, 0, 0},
				n:      15,
				offset: -1,
				r:      2.0,
			},
			want:  0.,
			want1: 1.0,
		},
		{
			name: "left",
			args: args{
				arr:    []byte{255, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0},
				n:      15,
				offset: -1,
				r:      2.0,
			},
			want:  -1.,
			want1: 1.0,
		},
		{
			name: "right",
			args: args{
				arr:    []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 255},
				n:      15,
				offset: -1,
				r:      2.0,
			},
			want:  1.,
			want1: 1.0,
		},
		{
			name: "right",
			args: args{
				arr:    []byte{0, 0, 0, 0, 0, 0, 0, 5, 10, 15, 20, 40, 100, 60, 5},
				n:      15,
				offset: -1,
				r:      2.0,
			},
			want:  0.7142857142857142,
			want1: 0.39215686274509803,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, got1 := LinearBin(tt.args.arr, tt.args.n, tt.args.offset, tt.args.r)
			if got != tt.want {
				t.Errorf("linearBin() got = %v, want %v", got, tt.want)
			}
			if got1 != tt.want1 {
				t.Errorf("linearBin() got1 = %v, want %v", got1, tt.want1)
			}
		})
	}
}

func TestParseModelType(t *testing.T) {
	type args struct {
		s string
	}
	tests := []struct {
		name string
		args args
		want ModelType
	}{
		{name: "categorical", args: args{s: "categorical"}, want: ModelTypeCategorical},
		{name: "categorical-upper", args: args{s: "caTeGorical"}, want: ModelTypeCategorical},
		{name: "linear", args: args{s: "linear"}, want: ModelTypeLinear},
		{name: "linear-upper", args: args{s: "LineAr"}, want: ModelTypeLinear},
		{name: "unknown", args: args{s: "1234"}, want: ModelTypeUnknown},
		{name: "empty", args: args{s: ""}, want: ModelTypeUnknown},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := ParseModelType(tt.args.s); got != tt.want {
				t.Errorf("ParseModelType() = %v, want %v", got, tt.want)
			}
		})
	}
}