lutra 0.13.3

Query runner for PRQL
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

#![cfg(not(target_family = "wasm"))]

use std::{path::PathBuf, str::FromStr};

use anyhow::Result;
use arrow::record_batch::RecordBatch;
use insta::{assert_debug_snapshot, assert_snapshot};
use itertools::Itertools;
use lutra::{DiscoverParams, ExecuteParams};
use prqlc::ir::pl::Ident;

fn example_project_params() -> DiscoverParams {
    DiscoverParams {
        project_path: PathBuf::from_str("../example-project").unwrap(),
    }
}

fn execute_example_project(
    expression_path: Option<String>,
) -> Result<Vec<(Ident, Vec<RecordBatch>)>> {
    let project = lutra::discover(example_project_params())?;

    let project = lutra::compile(project, Default::default())?;

    lutra::execute(project, ExecuteParams { expression_path })
}

#[test]
fn test_discover() {
    let project_tree = lutra::discover(example_project_params()).unwrap();

    let paths: Vec<_> = project_tree.sources.keys().sorted().collect();
    assert_debug_snapshot!(paths, @r#"
    [
        "Project.prql",
        "genres.prql",
    ]
    "#);
}

#[test]
fn test_execute() {
    let results = execute_example_project(Some("main".into())).unwrap();
    let (name, data) = results.into_iter().exactly_one().unwrap();

    assert_eq!(name.to_string(), "main");

    assert_snapshot!(arrow::util::pretty::pretty_format_batches(&data).unwrap(), @r"
    +-----+--------------+-------------+
    | aid | name         | last_listen |
    +-----+--------------+-------------+
    | 240 | Pink Floyd   | 2023-05-18  |
    | 14  | Apocalyptica | 2023-05-16  |
    +-----+--------------+-------------+
    ");
}

#[test]
fn test_missing() {
    let error = execute_example_project(Some("non_existent".into())).unwrap_err();

    assert_debug_snapshot!(error, @r#"
    Error {
        kind: Error,
        span: None,
        reason: Simple(
            "cannot find expression: `non_existent`",
        ),
        hints: [],
        code: None,
    }
    "#);
}