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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
|
import importlib.util
import unittest
import math
import platform
import zxingcpp
has_numpy = importlib.util.find_spec('numpy') is not None
has_pil = importlib.util.find_spec('PIL') is not None
has_cv2 = importlib.util.find_spec('cv2') is not None
BF = zxingcpp.BarcodeFormat
CT = zxingcpp.ContentType
class TestFormat(unittest.TestCase):
def test_format(self):
self.assertEqual(zxingcpp.barcode_format_from_str('qrcode'), BF.QRCode)
self.assertEqual(zxingcpp.barcode_formats_from_str('ITF, qrcode'), BF.ITF | BF.QRCode)
class TestReadWrite(unittest.TestCase):
def check_res(self, res, format, text):
self.assertTrue(res.valid)
self.assertEqual(res.format, format)
self.assertEqual(res.text, text)
self.assertEqual(res.bytes, bytes(text, 'utf-8'))
self.assertEqual(res.orientation, 0)
self.assertEqual(res.content_type, CT.Text)
def test_write_read_cycle(self):
format = BF.QRCode
text = "I have the best words."
img = zxingcpp.write_barcode(format, text)
res = zxingcpp.read_barcode(img)
self.check_res(res, format, text)
self.assertEqual(res.symbology_identifier, "]Q1")
# self.assertEqual(res.position.top_left.x, 4)
res = zxingcpp.read_barcode(img, formats=format)
self.check_res(res, format, text)
@unittest.skipIf(not hasattr(zxingcpp, 'create_barcode'), "skipping test for new create_barcode API")
def test_create_write_read_cycle(self):
format = BF.DataMatrix
text = "I have the best words."
img = zxingcpp.create_barcode(text, format).to_image()
res = zxingcpp.read_barcode(img)
self.check_res(res, format, text)
def test_write_read_oned_cycle(self):
format = BF.Code128
text = "I have the best words."
height = 50
width = 400
img = zxingcpp.write_barcode(format, text, width=width, height=height)
# self.assertEqual(img.shape[0], height)
# self.assertEqual(img.shape[1], width)
res = zxingcpp.read_barcode(img)
self.check_res(res, format, text)
# self.assertEqual(res.position.top_left.x, 61)
def test_write_read_multi_cycle(self):
format = BF.QRCode
text = "I have the best words."
img = zxingcpp.write_barcode(format, text)
res = zxingcpp.read_barcodes(img)[0]
self.check_res(res, format, text)
def test_write_read_bytes_cycle(self):
format = BF.QRCode
text = b"\1\2\3\4"
img = zxingcpp.write_barcode(format, text)
res = zxingcpp.read_barcode(img)
self.assertTrue(res.valid)
self.assertEqual(res.bytes, text)
self.assertEqual(res.content_type, CT.Binary)
@unittest.skipIf(not hasattr(zxingcpp, 'create_barcode'), "skipping test for new create_barcode API")
def test_create_write_read_bytes_cycle(self):
format = BF.DataMatrix
text = b"\1\2\3\4"
img = zxingcpp.create_barcode(text, format).to_image()
res = zxingcpp.read_barcode(img)
self.assertTrue(res.valid)
self.assertEqual(res.bytes, text)
self.assertEqual(res.content_type, CT.Binary)
@staticmethod
def zeroes(shape):
return memoryview(b"0" * math.prod(shape)).cast("B", shape=shape)
def test_failed_read_buffer(self):
res = zxingcpp.read_barcode(
self.zeroes((100, 100)), formats=BF.EAN8 | BF.Aztec, binarizer=zxingcpp.Binarizer.BoolCast
)
self.assertEqual(res, None)
@unittest.skipIf(not has_numpy, "need numpy for read/write tests")
def test_failed_read_numpy(self):
import numpy as np
res = zxingcpp.read_barcode(
np.zeros((100, 100), np.uint8), formats=BF.EAN8 | BF.Aztec, binarizer=zxingcpp.Binarizer.BoolCast
)
self.assertEqual(res, None)
def test_write_read_cycle_buffer(self):
format = BF.QRCode
text = "I have the best words."
img = zxingcpp.write_barcode(format, text)
self.check_res(zxingcpp.read_barcode(img), format, text)
@unittest.skipIf(not has_numpy or platform.system() == "Windows", "need numpy for read/write tests")
def test_write_read_cycle_numpy(self):
import numpy as np
format = BF.QRCode
text = "I have the best words."
img = zxingcpp.write_barcode(format, text, quiet_zone=10)
img = np.array(img)
self.check_res(zxingcpp.read_barcode(img), format, text)
self.check_res(zxingcpp.read_barcode(img[4:40,4:40]), format, text)
@unittest.skipIf(not has_pil, "need PIL for read/write tests")
def test_write_read_cycle_pil(self):
from PIL import Image
format = BF.QRCode
text = "I have the best words."
img = zxingcpp.write_barcode(format, text)
img = Image.fromarray(img, "L")
self.check_res(zxingcpp.read_barcode(img), format, text)
self.check_res(zxingcpp.read_barcode(img.convert("RGB")), format, text)
self.check_res(zxingcpp.read_barcode(img.convert("RGBA")), format, text)
self.check_res(zxingcpp.read_barcode(img.convert("1")), format, text)
self.check_res(zxingcpp.read_barcode(img.convert("CMYK")), format, text)
@unittest.skipIf(not has_cv2, "need cv2 for read/write tests")
def test_write_read_cycle_cv2(self):
import cv2, numpy
format = BF.QRCode
text = "I have the best words."
img = zxingcpp.write_barcode(format, text, quiet_zone=10)
img = cv2.cvtColor(numpy.array(img), cv2.COLOR_GRAY2BGR )
self.check_res(zxingcpp.read_barcode(img), format, text)
self.check_res(zxingcpp.read_barcode(img[4:40,4:40,:]), format, text)
def test_read_invalid_type(self):
self.assertRaisesRegex(
TypeError, "Invalid input: <class 'str'> does not support the buffer protocol.", zxingcpp.read_barcode, "foo"
)
def test_read_invalid_numpy_array_channels_buffer(self):
self.assertRaisesRegex(
ValueError, "Unsupported number of channels for buffer: 4", zxingcpp.read_barcode,
self.zeroes((100, 100, 4))
)
@unittest.skipIf(not has_numpy, "need numpy for read/write tests")
def test_read_invalid_numpy_array_channels_numpy(self):
import numpy as np
self.assertRaisesRegex(
ValueError, "Unsupported number of channels for buffer: 4", zxingcpp.read_barcode,
np.zeros((100, 100, 4), np.uint8)
)
if __name__ == '__main__':
unittest.main()
|
