fix some tests

Signed-off-by: Connor Tsui <connor.tsui20@gmail.com>

Author: connortsui20

encodings/bytebool/src/array.rs

@@ -352,7 +352,6 @@ impl From<Vec<Option<bool>>> for ByteBoolData {
 mod tests {
     use vortex_array::ArrayContext;
     use vortex_array::IntoArray;
-    use vortex_array::LEGACY_SESSION;
     use vortex_array::assert_arrays_eq;
     use vortex_array::serde::SerializeOptions;
     use vortex_array::serde::SerializedArray;
@@ -407,7 +406,7 @@ mod tests {
         let serialized = array
             .clone()
             .into_array()
-            .serialize(&ctx, &LEGACY_SESSION, &SerializeOptions::default())
+            .serialize(&ctx, &session, &SerializeOptions::default())
             .unwrap();
 
         let mut concat = ByteBufferMut::empty();

encodings/pco/src/test.rs

@@ -184,7 +184,7 @@ fn test_serde() -> VortexResult<()> {
     let bytes = pco
         .serialize(
             &context,
-            &LEGACY_SESSION,
+            &SESSION,
             &SerializeOptions {
                 offset: 0,
                 include_padding: true,

vortex-array/src/serde.rs

@@ -694,8 +694,6 @@ impl TryFrom<BufferHandle> for SerializedArray {
 
 #[cfg(test)]
 mod tests {
-    use std::sync::LazyLock;
-
     use flatbuffers::FlatBufferBuilder;
     use vortex_session::VortexSession;
     use vortex_session::registry::ReadContext;
@@ -710,8 +708,6 @@ mod tests {
     use crate::session::ArraySession;
     use crate::session::ArraySessionExt;
 
-    static SESSION: LazyLock<VortexSession> = LazyLock::new(VortexSession::empty);
-
     #[test]
     fn unknown_array_encoding_allow_unknown() {
         let mut fbb = FlatBufferBuilder::new();
@@ -770,11 +766,11 @@ mod tests {
             "vortex.test.foreign_child"
         );
         assert_eq!(
-            SESSION.array_serialize(&decoded).unwrap().unwrap(),
+            session.array_serialize(&decoded).unwrap().unwrap(),
             vec![1, 2, 3]
         );
         assert_eq!(
-            SESSION
+            session
                 .array_serialize(&decoded.nth_child(0).unwrap())
                 .unwrap()
                 .unwrap(),
@@ -784,7 +780,7 @@ mod tests {
         let serialized = decoded
             .serialize(
                 &ArrayContext::default(),
-                &SESSION,
+                &session,
                 &SerializeOptions::default(),
             )
             .unwrap();

vortex-array/src/session/mod.rs

@@ -15,6 +15,7 @@ use crate::arrays::Bool;
 use crate::arrays::Chunked;
 use crate::arrays::Constant;
 use crate::arrays::Decimal;
+use crate::arrays::Dict;
 use crate::arrays::Extension;
 use crate::arrays::FixedSizeList;
 use crate::arrays::List;
@@ -73,6 +74,7 @@ impl Default for ArraySession {
         // Register the utility encodings.
         this.register(Chunked);
         this.register(Constant);
+        this.register(Dict);
         this.register(List);
         this.register(Masked);
         this.register(Patched);

vortex-layout/src/layouts/flat/writer.rs

@@ -468,27 +468,23 @@ mod tests {
             let ctx = ArrayContext::empty();
 
             // Write the array into a byte buffer.
-            let (layout, _segments) = {
-                let segments = Arc::new(TestSegments::default());
-                let (ptr, eof) = SequenceId::root().split();
-                // Only allow the dict encoding; canonical primitive children remain permitted.
-                let mut allowed = HashSet::default();
-                allowed.insert(Dict.id());
-                let layout = FlatLayoutStrategy::default()
-                    .with_allow_encodings(allowed)
-                    .write_stream(
-                        ctx,
-                        Arc::<TestSegments>::clone(&segments),
-                        dict.into_array().to_array_stream().sequenced(ptr),
-                        eof,
-                        &session,
-                    )
-                    .await;
 
-                (layout, segments)
-            };
+            let segments = Arc::new(TestSegments::default());
+            let (ptr, eof) = SequenceId::root().split();
+            // Only allow the dict encoding; canonical primitive children remain permitted.
+            let mut allowed = HashSet::default();
+            allowed.insert(Dict.id());
 
-            assert!(layout.is_ok());
+            FlatLayoutStrategy::default()
+                .with_allow_encodings(allowed)
+                .write_stream(
+                    ctx,
+                    Arc::<TestSegments>::clone(&segments),
+                    dict.into_array().to_array_stream().sequenced(ptr),
+                    eof,
+                    &session,
+                )
+                .await?;
 
             Ok(())
         })