small refactor and cleanup

tests/test.py

@@ -0,0 +1,66 @@
+#!/usr/bin/python
+import sys
+import os
+import subprocess
+
+COLORS = {
+    "red": "\033[91m",
+    "green": "\033[92m",
+    "yellow": "\033[93m",
+    "blue": "\033[94m",
+    "reset": "\033[0m"
+}
+
+def print_colored(text, color):
+    print(COLORS.get(color, "") + text + COLORS["reset"], end="")
+
+def run_tests():
+    test_dir = "tests"
+    any_failed = False
+    
+    if not os.path.isdir(test_dir):
+        print("No 'tests' directory found.")
+        return 1
+
+    for file in sorted(os.listdir(test_dir)):
+        if file.endswith(".test"):
+            test_path = os.path.join(test_dir, file)
+            expected_path = test_path.replace(".test", ".expected")
+
+            if not os.path.isfile(expected_path):
+                print(f"Missing expected output file for {file}")
+                any_failed = True
+                continue
+
+            with open(test_path, "r") as test_file:
+                command = test_file.read().strip()
+
+            with open(expected_path, "r") as expected_file:
+                expected_output = expected_file.read().strip()
+
+            try:
+                result = subprocess.run(command, shell=True, text=True, capture_output=True)
+                actual_output = result.stdout.strip()
+                stderr_output = result.stderr.strip()
+
+                if result.returncode == 0 and actual_output == expected_output:
+                    print_colored("[OK] ", "green")
+                    print(f"{file} passed")
+                else:
+                    print_colored("[ERR] ", "red")
+                    print(f"{file} failed (exit {result.returncode})")
+                    print(f"Expected:\n{expected_output}")
+                    print(f"Got:\n{actual_output}")
+                    if stderr_output:
+                        print(f"Stderr:\n{stderr_output}")
+                    any_failed = True
+
+            except Exception as e:
+                print_colored(f"Error running {file}: {e}", "red")
+                any_failed = True
+
+    return 1 if any_failed else 0
+
+if __name__ == "__main__":
+    sys.exit(run_tests())
+