kattis-cli 1.0.4__py3-none-any.whl → 1.0.5__py3-none-any.whl

kattis_cli/main.py

@@ -7,7 +7,7 @@ build.sh script copies the contents of this file to main.py.
 Change the __version__ to match in pyproject.toml
 Has to be higher than the pypi version.
 """
-__version__ = '1.0.4'
+__version__ = '1.0.5'
 
 from math import inf
 from typing import Tuple

kattis_cli/test_solution.py

@@ -2,6 +2,7 @@
 """
 
 from typing import Any, List, Dict
+from math import inf
 import glob
 import time
 import os
@@ -26,7 +27,7 @@ def test_samples(
         problem_root_folder: str,
         files: List[str],
         lang_config: Dict[Any, Any],
-        accuracy: float = 0
+        accuracy: float = inf
 ) -> None:
     """Tests a problem by running all the .in files in
     the problem folder and comparing the output to the .ans files.
@@ -118,28 +119,31 @@ def test_samples(
                 input_content = f.read()
                 input_content.replace(b'\r\n', b'\n')  # Windows fix
             out_file = in_file.replace('.in', '.ans')
-            with open(out_file, 'rb') as f:
-                expected = f.read()
-                expected.replace(b'\r\n', b'\n')
+            try:
+                with open(out_file, 'rb') as f:
+                    expected = f.read()
+                    expected.replace(b'\r\n', b'\n')
+            except FileNotFoundError:
+                try:
+                    out_file = in_file.replace('.in', '.out')
+                    with open(out_file, 'rb') as f:
+                        expected = f.read()
+                        expected.replace(b'\r\n', b'\n')
+                except FileNotFoundError:
+                    expected = b"No .ans or .out file found!"
             # Run the program
             code, ans, error = run_program.run(lang_config, mainclass, in_file)
-            expected = expected.strip()
             if code != 0:
                 ans = error
             # console.print(f"{ans=} {error=}")
-            if accuracy == 0:  # string comparison
-                if expected == ans.encode('utf-8').strip():
-                    result = "[bold green]✅[/bold green]"
-                    count += 1
-                else:
-                    result = "[bold red]❌[/bold red]"
-            else:  # floating point comparison
-                if utility.compare_floats(expected.decode('utf-8'),
-                                          ans, accuracy):
-                    result = "[bold green]✅[/bold green]"
-                    count += 1
-                else:
-                    result = "[bold red]❌[/bold red]"
+
+            # floating point comparison
+            if utility.check_answer(expected.decode('utf-8'),
+                                    ans, accuracy):
+                result = "[bold green]✅[/bold green]"
+                count += 1
+            else:
+                result = "[bold red]❌[/bold red]"
 
             # UI Table Row ---
             in_filename = Path(in_file).parts[-1]

kattis_cli/utils/utility.py

@@ -2,6 +2,7 @@
 
 from pathlib import Path
 import os
+from math import inf
 from typing import Union
 import yaml
 
@@ -60,21 +61,28 @@ def find_problem_root_folder(
     raise FileNotFoundError("Error: Problem root folder not found.")
 
 
-def compare_floats(expected: str, ans: str, places: float) -> bool:
-    """Compare two floating point numbers with given accuracy.
+def check_answer(expected: str, ans: str, places: float = inf) -> bool:
+    """Compare two numeric strings with given precision.
 
     Args:
         expected (str): expected result
         ans (str): actual result
-        places (float): decimal places for approximation
+        places (float): decimal places for approximation;
+        default inf for string comparison
 
     Returns:
-        bool: True if the two numbers are equal within the given accuracy
+        bool: True if the two values are equal, False otherwise
     """
     expect_ans = expected.strip().split('\n')
     actual_ans = ans.strip().split('\n')
     if len(expect_ans) != len(actual_ans):
         return False
+
+    if places == inf:
+        if expect_ans == actual_ans:
+            return True
+        else:
+            return False
     try:
         for i, ex_ans in enumerate(expect_ans):
             flt_expected = float(ex_ans)

{kattis_cli-1.0.4.dist-info → kattis_cli-1.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kattis-cli
-Version: 1.0.4
+Version: 1.0.5
 Summary: A command-line tool for Kattis
 Home-page: https://github.com/rambasnet/kattis-cli
 Author: Ram Basnet

{kattis_cli-1.0.4.dist-info → kattis_cli-1.0.5.dist-info}/RECORD

@@ -3,17 +3,17 @@ kattis_cli/__init__.py,sha256=afN92pog2fGyicY6KNBofYbCBYbYj4Fpi_INUSpsc-E,402
 kattis_cli/download.py,sha256=kvUJdqRW3_ETvTd4XMruEs87RWPVHp_3a4VUhcYZEz8,5863
 kattis_cli/kattis.py,sha256=uZct7vjbEW2HA5FJr1ItSZ-b07P9JQHv5SccV4Pqjo8,13771
 kattis_cli/kattis_setup.py,sha256=rq_-Fz8oafxFJ8MOcxJqHQT8-bezgVHZXY2hm4EQF3U,4357
-kattis_cli/main.py,sha256=aI9EWMqAr-QawCzDUPJSpxwoIsEHSi2ed-23-KOeiAQ,4475
+kattis_cli/main.py,sha256=paMidWBPOCInt_o9WsyS9HoYdNIU346u-KTn9YB2Ymk,4475
 kattis_cli/settings.py,sha256=d5q4dYj9VqDSqPalleh2oZWtND-1bPB0T2IwdajFrBg,591
-kattis_cli/test_solution.py,sha256=g8LwVKx7Z1bSsXWs_JExCagBrrpKs7vL5PfaAoJHGKA,6114
+kattis_cli/test_solution.py,sha256=6lXaj9V5Vhfw95OywVls3Ekv86OvSYDrWI-s29YBdpk,6190
 kattis_cli/ui.py,sha256=dI06yncjm8sdLQkupPxGRODvQ6Gbawwuqn-67FrWg6I,2682
 kattis_cli/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kattis_cli/utils/config.py,sha256=6SIuiXA9sFADo8RZUj8A6rXb3uqmrWrmmArfZrqNQdw,2293
 kattis_cli/utils/languages.py,sha256=wlhl1Zng0gQAtsWW6FLxdkdzqeeY_qQ94wqa3NLypEU,8758
 kattis_cli/utils/run_program.py,sha256=NWQ6vtTeWgkaW75r91FIHGXR5cAbeu8yMb5hwzpYFsg,2613
-kattis_cli/utils/utility.py,sha256=XqiL7nVWCXs8DsoF6BqbkL0O9-348W-GKN1ScAzQdZ4,2732
-kattis_cli-1.0.4.dist-info/LICENSE,sha256=JmBa4SEKBCDWEgiOZcISU4tUCpli6xSpVlSYgkBXSNQ,1067
-kattis_cli-1.0.4.dist-info/METADATA,sha256=fs7fFZhce3ZgAMKeRSlMAYU2i_rVbyYN1LFDtAiUSek,6603
-kattis_cli-1.0.4.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-kattis_cli-1.0.4.dist-info/entry_points.txt,sha256=kyzGN20VqUPR_H0J_jJUKT-10-cAMFLVegQ6C7tbHss,47
-kattis_cli-1.0.4.dist-info/RECORD,,
+kattis_cli/utils/utility.py,sha256=iPmfsHi-DwYxP9mKU_VurNnLJXf0dngwYPxUdqckgYo,2907
+kattis_cli-1.0.5.dist-info/LICENSE,sha256=JmBa4SEKBCDWEgiOZcISU4tUCpli6xSpVlSYgkBXSNQ,1067
+kattis_cli-1.0.5.dist-info/METADATA,sha256=0WYu19PZ0_ih3Y49Uxgj1JnZRi23fYgoDFYGCUSWN6Q,6603
+kattis_cli-1.0.5.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+kattis_cli-1.0.5.dist-info/entry_points.txt,sha256=kyzGN20VqUPR_H0J_jJUKT-10-cAMFLVegQ6C7tbHss,47
+kattis_cli-1.0.5.dist-info/RECORD,,