Skip to content

Commit d432ce7

Browse files
committed
Use bi.itemsize to disambiguate "l" or "L"
1 parent 7f124bb commit d432ce7

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tests/test_buffers.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -47,12 +47,12 @@ def test_format_descriptor_format(cpp_name, expected_fmts, np_array_dtype):
4747
if np_array_dtype is not None:
4848
na = np.array([], dtype=np_array_dtype)
4949
bi = m.get_buffer_info(na)
50-
if fmt in ("i", "q"):
51-
assert bi.format in [fmt, "l"]
52-
elif fmt in ("I", "Q"):
53-
assert bi.format in [fmt, "L"]
54-
else:
55-
assert bi.format == fmt
50+
bif = bi.format
51+
if bif == "l":
52+
bif = "i" if bi.itemsize == 4 else "q"
53+
elif bif == "L":
54+
bif = "I" if bi.itemsize == 4 else "Q"
55+
assert bif == fmt
5656

5757

5858
def test_from_python():

0 commit comments

Comments
 (0)