From 38aa697d0daa567a1e1fa777ac09b370636345bc Mon Sep 17 00:00:00 2001 From: "Ralf W. Grosse-Kunstleve" Date: Thu, 18 May 2023 16:49:18 -0700 Subject: [PATCH] Fully address i|q|l ambiguity (hopefully). --- tests/test_buffers.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_buffers.py b/tests/test_buffers.py index c8fa69707..5b6970206 100644 --- a/tests/test_buffers.py +++ b/tests/test_buffers.py @@ -51,10 +51,10 @@ def test_format_descriptor_format(cpp_name, expected_fmts, np_array_dtype): # py::format_descriptor<> vs np.array: na = np.array([], dtype=np_array_dtype) bi = m.get_buffer_info(na) - if fmt == "q": - assert bi.format in ["q", "l"] - elif fmt == "Q": - assert bi.format in ["Q", "L"] + if fmt in ("i", "q"): + assert bi.format in [fmt, "l"] + elif fmt in ("I", "Q"): + assert bi.format in [fmt, "L"] else: assert bi.format == fmt