diff --git a/arrow/src/array/ffi.rs b/arrow/src/array/ffi.rs index 57329037bc4..12d6f440b78 100644 --- a/arrow/src/array/ffi.rs +++ b/arrow/src/array/ffi.rs @@ -74,6 +74,11 @@ mod tests { let result = &ArrayData::try_from(d1)?; assert_eq!(result, expected); + + unsafe { + Arc::from_raw(array); + Arc::from_raw(schema); + } Ok(()) } diff --git a/arrow/src/ffi.rs b/arrow/src/ffi.rs index 8eb28837ce5..84905af20a6 100644 --- a/arrow/src/ffi.rs +++ b/arrow/src/ffi.rs @@ -79,6 +79,8 @@ //! unsafe { //! Box::from_raw(out_array_ptr); //! Box::from_raw(out_schema_ptr); +//! Arc::from_raw(array_ptr); +//! Arc::from_raw(schema_ptr); //! } //! //! Ok(()) diff --git a/arrow/src/ffi_stream.rs b/arrow/src/ffi_stream.rs index bfc62b8888c..c06b5233344 100644 --- a/arrow/src/ffi_stream.rs +++ b/arrow/src/ffi_stream.rs @@ -500,6 +500,8 @@ mod tests { } assert_eq!(produced_batches, vec![batch.clone(), batch]); + + unsafe { Arc::from_raw(stream_ptr) }; Ok(()) } @@ -529,6 +531,8 @@ mod tests { } assert_eq!(produced_batches, vec![batch.clone(), batch]); + + unsafe { Arc::from_raw(stream_ptr) }; Ok(()) }