explicitly assert dtype of canonical array

spiraldb · Oct 28, 2024 · c854777 · c854777
1 parent 737a3c9
commit c854777
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/vortex-array/src/array/varbin/flatten.rs b/vortex-array/src/array/varbin/flatten.rs
@@ -30,7 +30,7 @@ mod test {
 
     use crate::array::varbin::builder::VarBinBuilder;
     use crate::validity::ArrayValidity;
-    use crate::IntoCanonical;
+    use crate::{ArrayDType, IntoCanonical};
 
     #[rstest]
     #[case(DType::Utf8(Nullability::Nullable))]
@@ -43,9 +43,10 @@ mod test {
         varbin.push_value("123456789012".as_bytes());
         // non-inlinable value
         varbin.push_value("1234567890123".as_bytes());
-        let varbin = varbin.finish(dtype);
+        let varbin = varbin.finish(dtype.clone());
 
         let canonical = varbin.into_canonical().unwrap().into_varbinview().unwrap();
+        assert_eq!(canonical.dtype(), &dtype);
 
         assert!(!canonical.is_valid(0));
         assert!(!canonical.is_valid(1));