Skip to content

Commit cb444be

Browse files
committed
test: cover CUDA e2e null-mask export
Make the cuDF e2e fixture export a nullable primitive column and validate that the host Arrow round-trip preserves its nulls. Keep the other fixture columns non-nullable so this test stays focused on the null-mask path that CI exercises through cuDF. Signed-off-by: Alexander Droste <alexander.droste@protonmail.com>
1 parent 25c0ff3 commit cb444be

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

  • vortex-test/e2e-cuda/src

vortex-test/e2e-cuda/src/lib.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ pub unsafe extern "C" fn export_array(
6868
) -> i32 {
6969
let mut ctx = CudaSession::create_execution_ctx(&SESSION).unwrap();
7070

71-
let primitive = PrimitiveArray::from_iter(0u32..5);
71+
let primitive = PrimitiveArray::from_option_iter([Some(0u32), None, Some(2), Some(3), None]);
7272
let decimal = DecimalArray::from_iter(0i128..5, DecimalDType::new(38, 2));
7373
let strings = VarBinViewArray::from_iter_str([
7474
"one",
@@ -124,7 +124,7 @@ pub unsafe extern "C" fn validate_array(
124124
let array = make_array(array_data);
125125
let struct_array = array.as_struct();
126126

127-
let primitive = UInt32Array::from_iter(0..5);
127+
let primitive = UInt32Array::from_iter([Some(0), None, Some(2), Some(3), None]);
128128
let decimal = Decimal128Array::from_iter_values(0..5)
129129
.with_precision_and_scale(38, 2)
130130
.expect("with_precision_and_scale");
@@ -138,7 +138,7 @@ pub unsafe extern "C" fn validate_array(
138138
let date = Date32Array::from(vec![100i32, 200, 300, 400, 500]);
139139

140140
let expected_fields = Fields::from_iter([
141-
Field::new("prims", primitive.data_type().clone(), false),
141+
Field::new("prims", primitive.data_type().clone(), true),
142142
Field::new("decimals", decimal.data_type().clone(), false),
143143
Field::new("strings", string.data_type().clone(), false),
144144
Field::new("dates", date.data_type().clone(), false),

0 commit comments

Comments
 (0)