ENH: Add zero_copy_only option check

apache · Oct 27, 2017 · e0561dc · e0561dc
1 parent de4ed3e
commit e0561dc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cpp/src/arrow/python/arrow_to_pandas.cc b/cpp/src/arrow/python/arrow_to_pandas.cc
@@ -968,6 +968,11 @@ class CategoricalBlock : public PandasBlock {
 
     if (data.num_chunks() == 1 && indices_first.null_count() == 0) {
       RETURN_NOT_OK(AllocateNDArrayFromIndices<T>(npy_type, indices_first));
+    } else if (options_.zero_copy_only) {
+      std::stringstream ss;
+      ss << "Needed to copy " << data.num_chunks() << " chunks with "
+         << indices_first.null_count() << " indices nulls, but zero_copy_only was True";
+      return Status::Invalid(ss.str());
     }
     else {
       RETURN_NOT_OK(AllocateNDArray(npy_type, 1));