def test_set_order_sparse(order, input_order): """Check that _set_order returns sparse matrices in promised format.""" X = sparse.coo_matrix(np.array([[0], [0], [0]])) y = sparse.coo_matrix(np.array([0, 0, 0])) sparse_format = "csc" if input_order == "F" else "csr" X = X.asformat(sparse_format) y = X.asformat(sparse_format) X2, y2 = _set_order(X, y, order=order) if order == 'C': assert sparse.isspmatrix_csr(X2) assert sparse.isspmatrix_csr(y2) elif order == 'F': assert sparse.isspmatrix_csc(X2) assert sparse.isspmatrix_csc(y2)
def test_set_order_dense(order, input_order): """Check that _set_order returns arrays with promised order.""" X = np.array([[0], [0], [0]], order=input_order) y = np.array([0, 0, 0], order=input_order) X2, y2 = _set_order(X, y, order=order) if order == 'C': assert X2.flags['C_CONTIGUOUS'] assert y2.flags['C_CONTIGUOUS'] elif order == 'F': assert X2.flags['F_CONTIGUOUS'] assert y2.flags['F_CONTIGUOUS'] if order == input_order: assert X is X2 assert y is y2