fix more tests

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

Author: connortsui20

encodings/zigzag/public-api.lock

@@ -42,6 +42,8 @@ pub fn vortex_zigzag::ZigZag::serialize(&self, _options: &vortex_array::scalar_f
 
 pub fn vortex_zigzag::ZigZag::validity(&self, _options: &vortex_array::scalar_fn::vtable::EmptyOptions, expression: &vortex_array::expr::expression::Expression) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::expression::Expression>>
 
+pub fn vortex_zigzag::initialize(session: &vortex_session::VortexSession)
+
 pub fn vortex_zigzag::zigzag_decode(parray: vortex_array::arrays::primitive::vtable::PrimitiveArray) -> vortex_array::arrays::primitive::vtable::PrimitiveArray
 
 pub fn vortex_zigzag::zigzag_encode(parray: vortex_array::array::view::ArrayView<'_, vortex_array::arrays::primitive::vtable::Primitive>) -> vortex_error::VortexResult<vortex_array::array::erased::ArrayRef>

vortex-array/public-api.lock

@@ -22084,8 +22084,6 @@ pub fn vortex_array::ArrayRef::is_valid(&self, index: usize) -> vortex_error::Vo
 
 pub fn vortex_array::ArrayRef::len(&self) -> usize
 
-pub fn vortex_array::ArrayRef::metadata(&self, session: &vortex_session::VortexSession) -> vortex_error::VortexResult<core::option::Option<alloc::vec::Vec<u8>>>
-
 pub fn vortex_array::ArrayRef::metadata_fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result
 
 pub fn vortex_array::ArrayRef::named_buffers(&self) -> alloc::vec::Vec<(alloc::string::String, vortex_array::buffer::BufferHandle)>

vortex-cuda/src/hybrid_dispatch/mod.rs

@@ -241,6 +241,7 @@ mod tests {
         let session = VortexSession::empty();
         fastlanes::initialize(&session);
         session.arrays().register(ZstdBuffers);
+
         let mut ctx = CudaSession::create_execution_ctx(&session).vortex_expect("ctx");
 
         let num_values: u32 = 64;
@@ -257,8 +258,7 @@ mod tests {
             0u32.into(),
         )
         .vortex_expect("for");
-        let vals = ZstdBuffers::compress(&vals.into_array(), 3, &VortexSession::empty())
-            .vortex_expect("zstd");
+        let vals = ZstdBuffers::compress(&vals.into_array(), 3, &session).vortex_expect("zstd");
 
         // codes = FoR(BitPacked)
         let codes = PrimitiveArray::new(

vortex-python/test/test_compress.py

@@ -32,13 +32,6 @@ def test_arrange_encode():
     assert compressed.nbytes < a.nbytes
 
 
-def test_zigzag_encode():
-    a = vortex.array(pa.array([-1, -1, 0, -1, 1, -1]))
-    zarr = vortex.ZigZagArray.encode(a)
-    assert isinstance(zarr, vortex.ZigZagArray)
-    # TODO(ngates): support decoding once we have decompressor.
-
-
 def test_chunked_encode():
     chunked = pa.chunked_array([pa.array([0, 1, 2]), pa.array([3, 4, 5])])
     encoded = vortex.array(chunked)