prevent accidental use of non-int typecodes

binref · Nov 28, 2023 · 50131e1 · 50131e1
1 parent c861c66
commit 50131e1
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/refinery/lib/chunks.py b/refinery/lib/chunks.py
@@ -11,9 +11,7 @@
 
 
 _BIG_ENDIAN = sys.byteorder == 'big'
-_TYPE_CODES = {
-    array.array(t).itemsize: t for t in array.typecodes if t.isupper()
-}
+_TYPE_CODES = {array.array(t).itemsize: t for t in 'BHILQ'}
 
 
 def unpack(data: bytes, blocksize: int, bigendian: bool = False, step: int = 0) -> Iterable[int]: