fix: tests all return code

Signed-off-by: Thing-han, Lim <[email protected]>

scripts/lib/mlkem_test.py

@@ -635,9 +635,8 @@ def bench(
     def all(self, func: bool, kat: bool, nistkat: bool):
         config_logger(self.verbose)
 
-        exit_code = 0
-
         def all(opt: bool):
+            code = 0
             if self.compile:
                 compiles = [
                     *([self._func.compile] if func else []),
@@ -649,7 +648,7 @@ def all(opt: bool):
                     try:
                         f(opt)
                     except SystemExit as e:
-                        exit_code = exit_code or e
+                        code = code or e
 
                     sys.stdout.flush()
 
@@ -664,13 +663,16 @@ def all(opt: bool):
                     try:
                         f(opt)
                     except SystemExit as e:
-                        exit_code = exit_code or e
+                        code = code or e
 
                     sys.stdout.flush()
+            return code
+
+        exit_code = 0
 
         if self.opt.lower() == "all" or self.opt.lower() == "no_opt":
-            all(False)
+            exit_code = exit_code or all(False)
         if self.opt.lower() == "all" or self.opt.lower() == "opt":
-            all(True)
+            exit_code = exit_code or all(True)
 
         exit(exit_code)