diff --git a/analysis/Cargo.toml b/analysis/Cargo.toml index 4f970f908..e82aa8661 100644 --- a/analysis/Cargo.toml +++ b/analysis/Cargo.toml @@ -16,6 +16,7 @@ type_check = { path = "../type_check" } [dev-dependencies] parser = { path = "../parser" } +importer = { path = "../importer" } pretty_assertions = "1.3.0" test-log = "0.2.12" env_logger = "0.10.0" diff --git a/analysis/src/lib.rs b/analysis/src/lib.rs index 758ac3d1a..641fd9bb8 100644 --- a/analysis/src/lib.rs +++ b/analysis/src/lib.rs @@ -62,11 +62,11 @@ pub mod utils { #[cfg(test)] mod test_util { use ast::asm_analysis::AnalysisASMFile; + use importer::resolve_str; use number::FieldElement; - use parser::parse_asm; /// A test utility to process a source file until after type checking pub fn typecheck_str(source: &str) -> Result, Vec> { - type_check::check(parse_asm(None, source).unwrap()) + type_check::check(resolve_str(source)) } } diff --git a/analysis/src/vm/batcher.rs b/analysis/src/vm/batcher.rs index 4340bbd8a..b5d182f8e 100644 --- a/analysis/src/vm/batcher.rs +++ b/analysis/src/vm/batcher.rs @@ -173,12 +173,12 @@ mod tests { let batched: AnalysisASMFile = batch_str(&input); - assert_eq!( - format!("{batched}") - .replace("\n\n", "\n") - .replace('\t', " "), - expected.replace("\n\n", "\n").replace('\t', " "), - ); + let result = format!("{batched}").replace("\n\n", "\n"); + let expected = expected.replace("\n\n", "\n"); + // The stdlib is added by the importer, so we only check for the initial part. + assert!(result.len() >= expected.len()); + assert!(expected.len() >= 50); + assert_eq!(result[..expected.len()], expected); } #[test]