refactor: trim whitespace in compiler only

kkrt-labs · Nov 6, 2024 · 16ec612 · 16ec612
1 parent 53fb010
commit 16ec612
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/crates/brainfuck_vm/src/bin/brainfuck_vm.rs b/crates/brainfuck_vm/src/bin/brainfuck_vm.rs
@@ -29,9 +29,7 @@ fn main() {
     // Constructs a subscriber whose severity level is filtered by `RUST_LOG`
     fmt().with_env_filter(EnvFilter::from_default_env()).init();
 
-    let code = fs::read_to_string(&args.filename)
-        .expect("Failed to read file")
-        .replace(' ', "");
+    let code = fs::read_to_string(&args.filename).expect("Failed to read file");
     let mut bf_compiler = Compiler::new(code);
     let ins = bf_compiler.compile();
     tracing::info!(

diff --git a/crates/brainfuck_vm/src/compiler.rs b/crates/brainfuck_vm/src/compiler.rs
@@ -54,6 +54,15 @@ mod tests {
         assert_eq!(expected_trimmed_code, compiler.code,);
     }
 
+    #[test]
+    fn test_whitespace() {
+        let code = " +  +> , < [> + .< - ]  ".to_string();
+        let compiler = Compiler::new(code);
+        let expected_trimmed_code =
+            vec!['+', '+', '>', ',', '<', '[', '>', '+', '.', '<', '-', ']'];
+        assert_eq!(expected_trimmed_code, compiler.code,);
+    }
+
     #[test]
     fn test_compile() {
         let code = "++>,<[>+.<-]".to_string();