python changes

Author: rok

python/pyarrow/tests/test_extension_type.py

@@ -26,7 +26,6 @@
 import pytest
 try:
     import numpy as np
-    from numpy.lib.stride_tricks import as_strided
 except ImportError:
     np = None
 import pyarrow as pa
@@ -1733,13 +1732,21 @@ def test_variable_shape_tensor_array_from_numpy(value_type):
         pa.VariableShapeTensorArray.from_numpy_ndarray([arr.astype(np.int32()), arr])
 
     flat_arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12], dtype=value_type)
+    bw = value_type.itemsize
 
     arr = flat_arr.reshape(1, 3, 4)
     tensor_array_from_numpy = pa.VariableShapeTensorArray.from_numpy_ndarray([arr])
     assert tensor_array_from_numpy.type.ndim == 3
     assert tensor_array_from_numpy.type.permutation == [0, 1, 2]
     assert tensor_array_from_numpy[0].to_tensor() == pa.Tensor.from_numpy(arr)
 
+    arr = as_strided(flat_arr, shape=(1, 2, 3, 2),
+                     strides=(bw * 12, bw * 6, bw, bw * 3))
+    tensor_array_from_numpy = pa.VariableShapeTensorArray.from_numpy_ndarray([arr])
+    assert tensor_array_from_numpy.type.ndim == 4
+    assert tensor_array_from_numpy.type.permutation == [0, 1, 3, 2]
+    assert tensor_array_from_numpy[0].to_tensor() == pa.Tensor.from_numpy(arr)
+
     arr = flat_arr.reshape(1, 2, 3, 2)
     result = pa.VariableShapeTensorArray.from_numpy_ndarray([arr])
     expected = np.array(