Test output as comment

src/tests/stdlib.rs

@@ -18,11 +18,32 @@ fn stdlib_test(input: &str) {
     let code =
         fs::read_to_string(input).unwrap_or_else(|_| panic!("Failed to open {input} test file"));
 
-    let output = match Path::new(&input.replace(".ab", ".output.txt")).exists() {
-        true => fs::read_to_string(input.replace(".ab", ".output.txt"))
-            .unwrap_or_else(|_| panic!("Failed to open {input}.output.txt file")),
-        _ => "Succeded".to_string(),
-    };
+    let mut is_output = false;
+    let mut output = "".to_owned();
+    for line in code.lines() {
+        if line.starts_with("// Output") {
+            is_output = true;
+            continue;
+        } else if line.is_empty() && is_output {
+            is_output = false;
+            break;
+        }
+
+        if is_output {
+            if ! output.is_empty() {
+                output.push_str("\n");
+            }
+            output.push_str(&line.replace("//", "").trim());
+        }
+    }
+
+    if output.is_empty() {
+        output = match Path::new(&input.replace(".ab", ".output.txt")).exists() {
+            true => fs::read_to_string(input.replace(".ab", ".output.txt"))
+            .expect(&format!("Failed to open {input}.output.txt file")),
+            _ => "Succeded".to_string()
+        };
+    }
 
     test_amber!(code, output);
 }

src/tests/stdlib/array_first_index.ab

@@ -1,4 +1,8 @@
 import * from "std/array"
+
+// Output
+// 2
+
 main {
     echo array_first_index([1, 2, 3, 4], 3)
 } 

src/tests/stdlib/array_search.ab

@@ -1,4 +1,8 @@
 import * from "std/array"
+
+// Output
+// 6
+
 main {
     let result = array_search([1, 2, 3, 4, 3], 3)
     echo result[0]+result[1]

src/tests/stdlib/chars.ab

@@ -1,4 +1,8 @@
 import * from "std/text"
+
+// Output
+// h e l l o
+
 main {
     echo chars("hello")
 } 

src/tests/stdlib/color_echo.ab

@@ -1,4 +1,8 @@
 import * from "std/env"
+
+// Output
+// Hello Amber!
+
 main {
     color_echo("Hello Amber!", 33)
 } 

src/tests/stdlib/lines.ab

@@ -1,4 +1,9 @@
 import { lines } from "std/text"
+
+// Output
+// line: hello
+// line: world
+
 main {
     loop line in lines("hello\nworld") {
         echo "line: " + line