Merge remote-tracking branch 'qubesos/pr/16'

* qubesos/pr/16:
  Fix off-by-one error in header length calculation
pull/18/head mm_dd71f295
Marek Marczykowski-Górecki 7 years ago
commit dd71f295e5
No known key found for this signature in database
GPG Key ID: 063938BA42CFA724

@ -49,9 +49,9 @@ ICON_MAXSIZE = 512
# header consists of two decimal numbers, SPC and LF
re_imghdr = re.compile(br'^\d+ \d+\n$')
imghdrlen = lambda w, h: int(math.ceil(math.log10(w)) \
+ math.ceil(math.log10(h)) \
+ 2)
def imghdrlen(w, h):
# width & height are inclusive max vals, and +2 for ' ' and '\n'
return len(str(w)) + len(str(h)) + 2
class Image(object):
def __init__(self, rgba, size):

@ -54,6 +54,7 @@ class TestCaseImage(unittest.TestCase):
class TestCaseFunctionsAndConstants(unittest.TestCase):
def test_00_imghdrlen(self):
self.assertEqual(qubesimgconverter.imghdrlen(8, 15), len('8 15\n'))
self.assertEqual(qubesimgconverter.imghdrlen(100, 100), len('100 100\n'))
def test_01_re_imghdr(self):
self.assertTrue(qubesimgconverter.re_imghdr.match('8 15\n'))

Loading…
Cancel
Save