1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

package detector

import "testing"

func TestDetectLicensePlate(t *testing.T) {
	tests := []struct {
		name      string
		input     string
		wantTexts []string
		wantConfs []float64
	}{
		{
			"standard hyphen",
			"B-AB 1234",
			[]string{"B-AB 1234"}, []float64{0.75},
		},
		{
			"three letter district",
			"MÜN-XY 99",
			[]string{"MÜN-XY 99"}, []float64{0.75},
		},
		{
			"e-vehicle suffix",
			"HH-AB 1234E",
			[]string{"HH-AB 1234E"}, []float64{0.75},
		},
		{
			"historic suffix",
			"K-CD 567H",
			[]string{"K-CD 567H"}, []float64{0.75},
		},
		{
			"space format lower confidence",
			"B AB 1234",
			[]string{"B AB 1234"}, []float64{0.65},
		},
		{
			"single letter district hyphen",
			"M-A 12",
			[]string{"M-A 12"}, []float64{0.75},
		},
		{
			"multiple",
			"B-AB 1234 und M-XY 999",
			[]string{"B-AB 1234", "M-XY 999"}, []float64{0.75, 0.75},
		},
		{
			"no match number starts with zero",
			"B-AB 0123",
			nil, nil,
		},
		{
			"no match too long district",
			"ABCD-XY 123",
			nil, nil,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := detectLicensePlate(tt.input)
			if len(got) != len(tt.wantTexts) {
				t.Fatalf("got %d findings, want %d — %v", len(got), len(tt.wantTexts), got)
			}
			for i, f := range got {
				if f.Text != tt.wantTexts[i] {
					t.Errorf("[%d] text = %q, want %q", i, f.Text, tt.wantTexts[i])
				}
				if f.Confidence != tt.wantConfs[i] {
					t.Errorf("[%d] confidence = %v, want %v", i, f.Confidence, tt.wantConfs[i])
				}
			}
		})
	}
}