python-ast 1.0.2

A library for compiling Python to Rust
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

use python_ast::{parse, CodeGen, CodeGenContext, PythonOptions, SymbolTableScopes};
use test_log::test;

#[test]
fn test_print_function_with_stdpython() {
    let code = r#"print("Hello, world!")"#;
    let module = parse(code, "test.py").unwrap();
    
    let options = PythonOptions::default();
    let rust_code = module.to_rust(
        CodeGenContext::Module("test".to_string()),
        options,
        SymbolTableScopes::new(),
    ).unwrap();
    
    let rust_str = rust_code.to_string();
    assert!(rust_str.contains("use stdpython :: *"));
    assert!(rust_str.contains("print"));
}

#[test]
fn test_basic_arithmetic_with_stdpython() {
    let code = r#"
result = 5 + 3
print(result)
"#;
    let module = parse(code, "test.py").unwrap();
    
    let options = PythonOptions::default();
    let rust_code = module.to_rust(
        CodeGenContext::Module("test".to_string()),
        options,
        SymbolTableScopes::new(),
    ).unwrap();
    
    let rust_str = rust_code.to_string();
    assert!(rust_str.contains("use stdpython :: *"));
}

#[test]
fn test_list_operations_with_stdpython() {
    let code = r#"
my_list = [1, 2, 3]
print(len(my_list))
"#;
    let module = parse(code, "test.py").unwrap();
    
    let options = PythonOptions::default();
    let rust_code = module.to_rust(
        CodeGenContext::Module("test".to_string()),
        options,
        SymbolTableScopes::new(),
    ).unwrap();
    
    let rust_str = rust_code.to_string();
    assert!(rust_str.contains("use stdpython :: *"));
    assert!(rust_str.contains("len"));
}

#[test]
fn test_function_definition_with_stdpython() {
    let code = r#"
def greet(name):
    return "Hello, " + name

result = greet("Python")
print(result)
"#;
    let module = parse(code, "test.py").unwrap();
    
    let options = PythonOptions::default();
    let rust_code = module.to_rust(
        CodeGenContext::Module("test".to_string()),
        options,
        SymbolTableScopes::new(),
    ).unwrap();
    
    let rust_str = rust_code.to_string();
    assert!(rust_str.contains("use stdpython :: *"));
}

#[test]
fn test_nostd_mode() {
    let code = r#"print("Hello, world!")"#;
    let module = parse(code, "test.py").unwrap();
    
    let mut options = PythonOptions::default();
    options.with_std_python = false;
    
    let rust_code = module.to_rust(
        CodeGenContext::Module("test".to_string()),
        options,
        SymbolTableScopes::new(),
    ).unwrap();
    
    let rust_str = rust_code.to_string();
    assert!(!rust_str.contains("use stdpython :: *"));
}