diff options
author | Ned Batchelder <ned@nedbatchelder.com> | 2022-12-27 09:16:18 -0500 |
---|---|---|
committer | Ned Batchelder <ned@nedbatchelder.com> | 2022-12-27 09:16:18 -0500 |
commit | 9afa02e8bb41c1351cfc0381c1a4dd0fac29e455 (patch) | |
tree | bfea30aa1e76d1273949a4389cf5ebd4f33a1bd4 /coverage/numbits.py | |
parent | b810bb130f81c0fbce43ae67244b10bc4699d930 (diff) | |
download | python-coveragepy-git-9afa02e8bb41c1351cfc0381c1a4dd0fac29e455.tar.gz |
refactor: don't need _to_blob anymore, leftover from 2 vs 3
Diffstat (limited to 'coverage/numbits.py')
-rw-r--r-- | coverage/numbits.py | 12 |
1 files changed, 4 insertions, 8 deletions
diff --git a/coverage/numbits.py b/coverage/numbits.py index e9ab1f90..da8e724b 100644 --- a/coverage/numbits.py +++ b/coverage/numbits.py @@ -19,10 +19,6 @@ from itertools import zip_longest from coverage.misc import contract -def _to_blob(b): - """Convert a bytestring into a type SQLite will accept for a blob.""" - return b - @contract(nums='Iterable', returns='blob') def nums_to_numbits(nums): @@ -38,11 +34,11 @@ def nums_to_numbits(nums): nbytes = max(nums) // 8 + 1 except ValueError: # nums was empty. - return _to_blob(b'') + return b'' b = bytearray(nbytes) for num in nums: b[num//8] |= 1 << num % 8 - return _to_blob(bytes(b)) + return bytes(b) @contract(numbits='blob', returns='list[int]') @@ -75,7 +71,7 @@ def numbits_union(numbits1, numbits2): A new numbits, the union of `numbits1` and `numbits2`. """ byte_pairs = zip_longest(numbits1, numbits2, fillvalue=0) - return _to_blob(bytes(b1 | b2 for b1, b2 in byte_pairs)) + return bytes(b1 | b2 for b1, b2 in byte_pairs) @contract(numbits1='blob', numbits2='blob', returns='blob') @@ -87,7 +83,7 @@ def numbits_intersection(numbits1, numbits2): """ byte_pairs = zip_longest(numbits1, numbits2, fillvalue=0) intersection_bytes = bytes(b1 & b2 for b1, b2 in byte_pairs) - return _to_blob(intersection_bytes.rstrip(b'\0')) + return intersection_bytes.rstrip(b'\0') @contract(numbits1='blob', numbits2='blob', returns='bool') |