mirror of https://github.com/mtshiba/pylyzer
73 lines
1.6 KiB
Rust
73 lines
1.6 KiB
Rust
use std::path::PathBuf;
|
|
|
|
use erg_common::config::{ErgConfig, Input};
|
|
use erg_common::spawn::exec_new_thread;
|
|
use erg_common::traits::Stream;
|
|
use erg_compiler::artifact::{CompleteArtifact, IncompleteArtifact};
|
|
use pylyzer::PythonAnalyzer;
|
|
|
|
pub fn exec_analyzer(file_path: &'static str) -> Result<CompleteArtifact, IncompleteArtifact> {
|
|
let cfg = ErgConfig {
|
|
input: Input::File(PathBuf::from(file_path)),
|
|
..Default::default()
|
|
};
|
|
let mut analyzer = PythonAnalyzer::new(cfg);
|
|
let py_code = analyzer.cfg.input.read();
|
|
analyzer.analyze(py_code, "exec")
|
|
}
|
|
|
|
fn _expect(file_path: &'static str, warns: usize, errors: usize) {
|
|
println!("Testing {file_path} ...");
|
|
match exec_analyzer(file_path) {
|
|
Ok(artifact) => {
|
|
assert_eq!(artifact.warns.len(), warns);
|
|
assert_eq!(errors, 0);
|
|
}
|
|
Err(artifact) => {
|
|
assert_eq!(artifact.warns.len(), warns);
|
|
assert_eq!(artifact.errors.len(), errors);
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn expect(file_path: &'static str, warns: usize, errors: usize) {
|
|
exec_new_thread(move || {
|
|
_expect(file_path, warns, errors);
|
|
});
|
|
}
|
|
|
|
#[test]
|
|
fn exec_test() {
|
|
expect("tests/test.py", 0, 10);
|
|
}
|
|
|
|
#[test]
|
|
fn exec_import() {
|
|
expect("tests/import.py", 0, 2);
|
|
}
|
|
|
|
#[test]
|
|
fn exec_export() {
|
|
expect("tests/export.py", 0, 0);
|
|
}
|
|
|
|
#[test]
|
|
fn exec_func() {
|
|
expect("tests/func.py", 0, 1);
|
|
}
|
|
|
|
#[test]
|
|
fn exec_class() {
|
|
expect("tests/class.py", 0, 4);
|
|
}
|
|
|
|
#[test]
|
|
fn exec_errors() {
|
|
expect("tests/errors.py", 0, 3);
|
|
}
|
|
|
|
#[test]
|
|
fn exec_warns() {
|
|
expect("tests/warns.py", 2, 0);
|
|
}
|