Fix initialized sessions in encoding tests

Signed-off-by: Nicholas Gates <nick@nickgates.com>

Author: gatesn

encodings/parquet-variant/src/kernel.rs

@@ -251,10 +251,17 @@ mod tests {
     use vortex_error::vortex_ensure;
     use vortex_error::vortex_err;
     use vortex_mask::Mask;
+    use vortex_session::VortexSession;
 
     use crate::ParquetVariant;
     use crate::ParquetVariantArrayExt;
 
+    fn session() -> VortexSession {
+        let session = vortex_array::array_session();
+        crate::initialize(&session);
+        session
+    }
+
     fn make_unshredded_array() -> VortexResult<ArrayRef> {
         let mut builder = VariantArrayBuilder::new(4);
         builder.append_variant(PqVariant::from(42i32));
@@ -375,9 +382,10 @@ mod tests {
         dtype: Option<VortexDType>,
     ) -> VortexResult<ArrayRef> {
         let expr = variant_get(root(), parse_path(path)?, dtype);
+        let session = session();
         array
             .apply(&expr)?
-            .execute::<ArrayRef>(&mut LEGACY_SESSION.create_execution_ctx())
+            .execute::<ArrayRef>(&mut session.create_execution_ctx())
     }
 
     macro_rules! assert_rows_eq {
@@ -826,7 +834,8 @@ mod tests {
         array: &ArrayRef,
         expected: impl IntoIterator<Item = Option<i32>>,
     ) -> VortexResult<()> {
-        let mut ctx = LEGACY_SESSION.create_execution_ctx();
+        let session = session();
+        let mut ctx = session.create_execution_ctx();
         let executed = array.clone().execute::<ArrayRef>(&mut ctx)?;
         let typed_value = executed
             .as_::<ParquetVariant>()

encodings/runend/src/compute/filter.rs

@@ -116,20 +116,27 @@ fn filter_run_end_primitive<R: NativePType + AddAssign + From<bool> + AsPrimitiv
 #[cfg(test)]
 mod tests {
     use vortex_array::IntoArray;
-    use vortex_array::LEGACY_SESSION;
     use vortex_array::VortexSessionExecute;
     use vortex_array::arrays::PrimitiveArray;
     use vortex_array::assert_arrays_eq;
     use vortex_error::VortexResult;
     use vortex_mask::Mask;
+    use vortex_session::VortexSession;
 
     use crate::RunEnd;
     use crate::RunEndArray;
 
+    fn session() -> VortexSession {
+        let session = vortex_array::array_session();
+        crate::initialize(&session);
+        session
+    }
+
     fn ree_array() -> RunEndArray {
+        let session = session();
         RunEnd::encode(
             PrimitiveArray::from_iter([1, 1, 1, 4, 4, 4, 2, 2, 5, 5, 5, 5]).into_array(),
-            &mut LEGACY_SESSION.create_execution_ctx(),
+            &mut session.create_execution_ctx(),
         )
         .unwrap()
     }
@@ -139,7 +146,8 @@ mod tests {
         let arr = ree_array().slice(2..7)?;
         let filtered = arr.filter(Mask::from_iter([true, false, false, true, true]))?;
 
-        let mut ctx = LEGACY_SESSION.create_execution_ctx();
+        let session = session();
+        let mut ctx = session.create_execution_ctx();
         assert_arrays_eq!(
             filtered,
             RunEnd::new(
@@ -157,7 +165,8 @@ mod tests {
     /// Filter unwrap one layer at a time so RunEnd's FilterKernel can fire.
     #[test]
     fn filter_sliced_run_end_preserves_encoding() -> VortexResult<()> {
-        let mut ctx = LEGACY_SESSION.create_execution_ctx();
+        let session = session();
+        let mut ctx = session.create_execution_ctx();
 
         // 4 runs of 32 each = 128 rows. Large enough that FilterKernel takes
         // the run-preserving path (true_count >= 25).