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
|
#
# Copyright (c) 2016 - 2024 -- Lars Heuer
# All rights reserved.
#
# License: BSD License
#
"""\
Utility functions for test cases.
"""
import os
def matrix_looks_valid(matrix):
"""\
Returns if the matrix contains just ``0x0`` and ``0x1`` values.
Does not check if the matrix represents a valid (Micro) QR Code.
:param matrix: tuple of bytearrays
:return:
"""
invalid_values = []
for i, row in enumerate(matrix):
for j, bit in enumerate(row):
if bit not in (0x0, 0x1):
invalid_values.append((i, j, bit))
if not invalid_values:
return True, ''
msg = 'Invalid values: '
for i, j, bit in invalid_values:
msg += f'\nrow: {i}, col {j}, value: {bit}'
return False, msg
def read_matrix(name):
"""\
Helper function to read a matrix from /ref_matrix. The file extension .txt
is added automatically.
:return: A tuple of bytearrays
"""
matrix = []
with open(os.path.join(os.path.dirname(__file__), f'ref_matrix/{name}.txt')) as f:
for row in f:
matrix.append(bytearray([int(i) for i in row if i != '\n']))
return tuple(matrix), len(matrix[0]), len(matrix[1])
|
