Implementing review feedback.

Author: rok

python/pyarrow/tensor.pxi

@@ -203,7 +203,7 @@ shape: {0.shape}""".format(self)
         coords = np.require(coords, dtype='i8', requirements='F')
 
         check_status(NdarraysToSparseCOOTensor(c_default_memory_pool(),
-                     obj.data.view(), coords, c_shape, c_dim_names,
+                     obj.data, coords, c_shape, c_dim_names,
                      &csparse_tensor))
         return pyarrow_wrap_sparse_coo_tensor(csparse_tensor)
 
@@ -227,10 +227,10 @@ shape: {0.shape}""".format(self)
             for x in dim_names:
                 c_dim_names.push_back(tobytes(x))
 
-        coords = np.require(obj.coords.T.view(), dtype='i8', requirements='F')
+        coords = np.require(obj.coords.T, dtype='i8', requirements='F')
 
         check_status(NdarraysToSparseCOOTensor(c_default_memory_pool(),
-                     obj.data.view(), coords, c_shape, c_dim_names,
+                     obj.data, coords, c_shape, c_dim_names,
                      &csparse_tensor))
         return pyarrow_wrap_sparse_coo_tensor(csparse_tensor)
 
@@ -286,7 +286,7 @@ shape: {0.shape}""".format(self)
                                               &out_data, &out_coords))
         data = PyObject_to_object(out_data)
         coords = PyObject_to_object(out_coords)
-        result = COO(data=data, coords=coords.T, shape=self.shape)
+        result = COO(data=data[:, 0], coords=coords.T, shape=self.shape)
         return result
 
     def to_tensor(self):

python/pyarrow/tests/test_sparse_tensor.py

@@ -336,7 +336,7 @@ def test_sparse_csr_matrix_scipy_roundtrip(dtype_str, arrow_type):
 @pytest.mark.parametrize('dtype_str,arrow_type', tensor_type_pairs)
 def test_pydata_sparse_sparse_coo_tensor_roundtrip(dtype_str, arrow_type):
     dtype = np.dtype(dtype_str)
-    data = np.array([[1, 2, 3, 4, 5, 6]]).T.astype(dtype)
+    data = np.array([1, 2, 3, 4, 5, 6]).astype(dtype)
     coords = np.array([
         [0, 0, 2, 3, 1, 3],
         [0, 2, 0, 4, 5, 5],
@@ -354,3 +354,5 @@ def test_pydata_sparse_sparse_coo_tensor_roundtrip(dtype_str, arrow_type):
     assert sparse_array.dtype == out_sparse_array.dtype
     assert np.array_equal(sparse_array.data, out_sparse_array.data)
     assert np.array_equal(sparse_array.coords, out_sparse_array.coords)
+    assert np.array_equal(sparse_array.todense(),
+                          sparse_tensor.to_tensor().to_numpy())