fixed bug

JeanExtreme002 · Apr 20, 2024 · ac8ab04 · ac8ab04
1 parent 8b3eed6
commit ac8ab04
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/fastsnake/__init__.py b/fastsnake/__init__.py
@@ -2,7 +2,7 @@
 A Python Helper CLI for Competitive Programming
 """
 
-__version__ = "1.4.8"
+__version__ = "1.4.9"
 
 __author__ = "Jean Loui Bernard Silva de Jesus"
 __github__ = "https://github.com/JeanExtreme002"

diff --git a/fastsnake/application/runner.py b/fastsnake/application/runner.py
@@ -90,6 +90,8 @@ def run_test(problem: str, step_counter: bool = False, debug: bool = False) -> b
             result, sc_result = result.split(f"{step_counter_variable}:")
             step_counter_result = int(sc_result.replace(f":{step_counter_variable}", ""))
 
+        result = result.rstrip("\n")
+
         # Load the expected output.
         output_filename = os.path.abspath(os.path.join(config["test_cases_namespace"], filename[:-2] + "out"))
 
@@ -204,6 +206,8 @@ def run_test_generator(problem: str, tests: int = 1, step_counter: bool = False,
             result, sc_result = result.split(f"{step_counter_variable}:")
             step_counter_result = int(sc_result.replace(f":{step_counter_variable}", ""))
 
+        result = result.rstrip("\n")
+
         # Check the output.
         try:
             check = generator.test_output(input_data, result)