ar_row 1.3.0

Row-oriented access to Arrow arrays
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

// Copyright (C) 2024 The Software Heritage developers
// See the AUTHORS file at the top-level directory of this distribution
// License: GNU General Public License version 3, or any later version
// See top-level LICENSE file for more information

use std::sync::Arc;

use ar_row::arrow::array::{Array, DictionaryArray, Int8Array};
use ar_row::arrow::datatypes::Int8Type;

use ar_row::deserialize::ArRowDeserialize;

#[test]
fn test_utf8_dict() {
    let array: DictionaryArray<Int8Type> = vec!["a", "a", "b", "c", "a"].into_iter().collect();
    assert_eq!(
        array.keys(),
        &Int8Array::from(vec![Some(0), Some(0), Some(1), Some(2), Some(0)])
    );

    let array: Arc<dyn Array> = Arc::new(array);

    let rows: Vec<_> = <String>::from_array(array).unwrap();

    assert_eq!(
        rows,
        vec![
            "a".to_string(),
            "a".to_string(),
            "b".to_string(),
            "c".to_string(),
            "a".to_string()
        ]
    );
}

#[test]
fn test_utf8_dict_opt() {
    let array: DictionaryArray<Int8Type> = vec![Some("a"), None, Some("b"), Some("c"), Some("a")]
        .into_iter()
        .collect();
    assert_eq!(
        array.keys(),
        &Int8Array::from(vec![Some(0), None, Some(1), Some(2), Some(0)])
    );

    let array: Arc<dyn Array> = Arc::new(array);

    let rows: Vec<_> = <Option<String>>::from_array(array.clone()).unwrap();

    assert_eq!(
        rows,
        vec![
            Some("a".to_string()),
            None,
            Some("b".to_string()),
            Some("c".to_string()),
            Some("a".to_string())
        ]
    );

    assert!(<String>::from_array(array).is_err());
}