polite 0.1.3

rusqlite × Polars bridge library
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

// polite/tests/roundtrip.rs
use polars::prelude::*;
use polite::prelude::*;
use tempfile::NamedTempFile;

#[test]
fn test_roundtrip_dataframe() {
    // Temp DB on disk (ConnectorX needs file-based DB)
    let db = NamedTempFile::new().unwrap();
    let db_path = db.path().to_str().unwrap();

    // Build a Polars DataFrame
    let df = df! {
        "id" => &[1i64, 2, 3],
        "name" => &["Alice", "Bob", "Charlie"],
    }
    .unwrap();

    // Insert into SQLite (rusqlite path)
    let conn = connect_sqlite(Some(db_path)).unwrap();
    from_dataframe(&conn, "people", &df).unwrap();

    // Read back using ConnectorX → Polars
    let df2 = to_dataframe(db_path, "SELECT * FROM people ORDER BY id").unwrap();

    // Check shapes
    assert_eq!(df.shape(), df2.shape());

    // Check values
    assert_eq!(
        df.column("id")
            .unwrap()
            .i64()
            .unwrap()
            .into_no_null_iter()
            .collect::<Vec<_>>(),
        df2.column("id")
            .unwrap()
            .i64()
            .unwrap()
            .into_no_null_iter()
            .collect::<Vec<_>>()
    );
    assert_eq!(
        df.column("name")
            .unwrap()
            .str()
            .unwrap()
            .into_no_null_iter()
            .collect::<Vec<_>>(),
        df2.column("name")
            .unwrap()
            .str()
            .unwrap()
            .into_no_null_iter()
            .collect::<Vec<_>>()
    );
}

#[test]
#[ignore] // Values not coming through correctly yet
fn test_roundtrip_dataframe_values() {
    let db = NamedTempFile::new().unwrap();
    let db_path = db.path().to_str().unwrap();
    let conn = connect_sqlite(Some(db_path)).unwrap();

    // Build a Polars DataFrame
    let df = df! {
        "id" => &[1, 2, 3],
        "name" => &["Alice", "Bob", "Charlie"],
    }
    .unwrap();

    // Insert into SQLite
    from_dataframe(&conn, "people", &df).unwrap();

    // Query back
    let df2 = to_dataframe(db_path, "SELECT * FROM people ORDER BY id").unwrap();

    // Compare values
    assert_eq!(df.shape(), df2.shape());
    assert_eq!(
        df.column("id")
            .unwrap()
            .i64()
            .unwrap()
            .into_no_null_iter()
            .collect::<Vec<_>>(),
        df2.column("id")
            .unwrap()
            .i64()
            .unwrap()
            .into_no_null_iter()
            .collect::<Vec<_>>()
    );
    assert_eq!(
        df.column("name")
            .unwrap()
            .str()
            .unwrap()
            .into_no_null_iter()
            .collect::<Vec<_>>(),
        df2.column("name")
            .unwrap()
            .str()
            .unwrap()
            .into_no_null_iter()
            .collect::<Vec<_>>()
    );
}