use growable api for test

jorgecarleitao · Nov 7, 2021 · 0517b95 · 0517b95
1 parent 316c26c
commit 0517b95
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/it/array/growable/dictionary.rs b/tests/it/array/growable/dictionary.rs
@@ -43,9 +43,10 @@ fn test_negative_keys() {
 
     let arr = DictionaryArray::from_data(keys, Arc::new(Utf8Array::<i32>::from(vals)));
     // check that we don't panic with negative keys to usize conversion
-    let out = arrow2::compute::concat::concatenate(&[&arr]).unwrap();
-    let out = out.as_any().downcast_ref::<DictionaryArray<i32>>().unwrap();
-    assert_eq!(out, &arr);
+    let mut growable = GrowableDictionary::new(&[&arr], false, 0);
+    growable.extend(0, 0, 4);
+    let out: DictionaryArray<i32> = growable.into();
+    assert_eq!(out, arr);
 }
 
 #[test]