1.3.1 RC 1

Author: flexatone

README.rst

@@ -35,6 +35,12 @@ ArrayKit requires the following:
 What is New in ArrayKit
 -------------------------
 
+1.3.1
+............
+
+Improved ``slice_to_unit()`` integer extraction.
+
+
 1.3.0
 ............
 

VERSION

@@ -1,2 +1,2 @@
-1.3.0
+1.3.1
 

src/methods.c

@@ -82,7 +82,7 @@ slice_to_unit(PyObject *Py_UNUSED(m), PyObject *a)
 
     Py_ssize_t step = 1;
     if (py_step != Py_None) {
-        step = PyLong_AsSsize_t(py_step);
+        step = PyNumber_AsSsize_t(py_step, NULL);
         if (step == -1 && PyErr_Occurred()) {
             return NULL;
         }
@@ -93,12 +93,12 @@ slice_to_unit(PyObject *Py_UNUSED(m), PyObject *a)
 
     Py_ssize_t start = 0;
     if (py_start != Py_None) {
-        start = PyLong_AsSsize_t(py_start);
+        start = PyNumber_AsSsize_t(py_start, NULL);
         if (start == -1 && PyErr_Occurred()) {
             return NULL;
         }
     }
-    Py_ssize_t stop = PyLong_AsSsize_t(py_stop);
+    Py_ssize_t stop = PyNumber_AsSsize_t(py_stop, NULL);
     if (stop == -1 && PyErr_Occurred()) {
         return NULL;
     }

test/test_util.py

@@ -972,6 +972,13 @@ def test_slice_to_unit_c(self) -> None:
         with self.assertRaises(TypeError):
             _ = slice_to_unit(3)
 
+    def test_slice_to_unit_d(self) -> None:
+        # __index__-able (NumPy integer) bounds are accepted
+        self.assertEqual(slice_to_unit(slice(np.int64(3), np.int64(4))), 3)
+        self.assertEqual(slice_to_unit(slice(np.int64(2), np.int64(4))), -1)
+        self.assertEqual(slice_to_unit(slice(None, np.int64(1))), 0)
+        self.assertEqual(slice_to_unit(slice(np.int64(2), np.int64(3), np.int64(1))), 2)
+
 
 if __name__ == '__main__':
     unittest.main()