rs-flint 0.1.11

RustScript-native AI inference with Torch, llama.cpp, and stable-diffusion.cpp
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

use std::path::PathBuf;
use std::sync::Arc;

use flint_ai::{ScriptRunner, compile_script_file};

#[test]
fn cli_scripts_compile_with_host_argparse() {
    let root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("scripts");
    for name in [
        "flux_klein.rss",
        "flux_klein_encode_prompt.rss",
        "ggml_devices.rss",
        "ggml_devices_from_path.rss",
        "lfm2.rss",
        "lfm2_5.rss",
        "llama_quantized.rss",
        "xlm_roberta_ner_japanese.rss",
    ] {
        let path = root.join(name);
        compile_script_file(&path)
            .unwrap_or_else(|error| panic!("failed to compile {}: {error}", path.display()));
    }
}

#[test]
fn cli_host_parses_typed_options() {
    let path = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("tests/fixtures/cli.rss");
    let program = Arc::new(compile_script_file(path).unwrap().program);
    let output = ScriptRunner::new()
        .run_text(
            program,
            vec![
                "--name=demo".to_owned(),
                "--count".to_owned(),
                "7".to_owned(),
                "--no-mmap".to_owned(),
            ],
        )
        .unwrap();
    assert_eq!(output.text, "demo");
}

#[test]
fn cli_host_rejects_missing_required_options() {
    let path = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("tests/fixtures/cli.rss");
    let program = Arc::new(compile_script_file(path).unwrap().program);
    let error = match ScriptRunner::new().run_text(program, Vec::new()) {
        Ok(_) => panic!("missing required option was accepted"),
        Err(error) => error,
    };
    assert!(
        error
            .to_string()
            .contains("required argument '-n' is missing")
    );
}

#[test]
fn diffusion_namespace_is_bound_at_runtime() {
    let path = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("tests/fixtures/diffusion.rss");
    let program = Arc::new(compile_script_file(path).unwrap().program);
    let output = ScriptRunner::new().run_text(program, Vec::new()).unwrap();
    assert_eq!(output.text, "euler:simple");
}