Mercurial > hgrepos > Python2 > PyMuPDF
diff mupdf-source/thirdparty/zxing-cpp/wrappers/python/test.py @ 2:b50eed0cc0ef upstream
ADD: MuPDF v1.26.7: the MuPDF source as downloaded by a default build of PyMuPDF 1.26.4.
The directory name has changed: no version number in the expanded directory now.
| author | Franz Glasner <fzglas.hg@dom66.de> |
|---|---|
| date | Mon, 15 Sep 2025 11:43:07 +0200 |
| parents | |
| children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mupdf-source/thirdparty/zxing-cpp/wrappers/python/test.py Mon Sep 15 11:43:07 2025 +0200 @@ -0,0 +1,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, "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()
