kattis-cli 1.0.1__py3-none-any.whl → 1.0.2__py3-none-any.whl

kattis_cli/main.py

@@ -1,6 +1,7 @@
 """ Main module for the kattis_cli package."""
-__version__ = '0.1.6'
+__version__ = '1.0.2'
 
+from math import inf
 from typing import Tuple
 from rich.console import Console
 import click
@@ -52,11 +53,14 @@ def info(problemid: str) -> None:
 @click.option('-p', '--problemid', default='', help='Problem ID')
 @click.option('-l', '--language', default='', help='Sets language')
 @click.option('-m', '--mainclass', default='', help='Sets mainclass/mainfile')
+@click.option('-a', '--accuracy', default=inf,
+              help='Decimal places for float comparison')
 @click.argument('files', nargs=-1, required=False)
 def test(
         problemid: str,
         language: str,
         mainclass: str,
+        accuracy: float,
         files: Tuple[str]) -> None:
     """Test solution with sample files.
     """
@@ -74,7 +78,8 @@ def test(
         mainclass,
         root_folder,
         _files,
-        lang_config)
+        lang_config,
+        accuracy)
 
 
 @main.command(help='Submit a solution to Kattis.')

kattis_cli/test_solution.py

@@ -1,6 +1,7 @@
 """Tester module for Kattis.
 """
 
+from math import inf
 from typing import Any, List, Dict
 import glob
 import time
@@ -19,13 +20,33 @@ from kattis_cli import kattis
 from kattis_cli.utils import run_program, utility
 
 
+def compare_floats(expected: str, ans: str, places: float) -> bool:
+    """Compare two floating point numbers with given accuracy.
+
+    Args:
+        expected (str): expected result
+        ans (str): actual result
+        places (float): decimal places for approximation
+
+    Returns:
+        bool: True if the two numbers are equal within the given accuracy
+    """
+    try:
+        flt_expected = float(expected)
+        flt_ans = float(ans)
+        return abs(flt_expected - flt_ans) <= 10**(-places)
+    except ValueError:
+        return False
+
+
 def test_samples(
         problemid: str,
         loc_language: str,
         mainclass: str,
         problem_root_folder: str,
         files: List[str],
-        lang_config: Dict[Any, Any]
+        lang_config: Dict[Any, Any],
+        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.
@@ -126,11 +147,18 @@ def test_samples(
             if code != 0:
                 ans = error
             # console.print(f"{ans=} {error=}")
-            if expected == ans.encode('utf-8').strip():
-                result = "[bold green]✅[/bold green]"
-                count += 1
-            else:
-                result = "[bold red]❌[/bold red]"
+            if accuracy == inf:  # 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 compare_floats(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

@@ -66,7 +66,6 @@ GUESS_MAINFILE = {
     'Common Lisp',
     'Pascal',
     'PHP',
-    'Python 2',
     'Python 3',
     'Ruby',
     'Rust',
@@ -77,7 +76,6 @@ GUESS_MAINFILE = {
 # mapping is used for .kattis-cli.toml file configuration
 LOCAL_TO_KATTIS = {
     'python3': 'Python 3',
-    'python2': 'Python 2',
     'java': 'Java',
     'cpp': 'C++',
     'c++': 'C++',
@@ -132,40 +130,10 @@ def guess_language(ext: str, files: List[str]) -> str:
         else:
             return "cpp"
     if ext == ".py":
-        if is_python2(files):
-            return "python2"
-        else:
-            return "python3"
+        return "python3"
     return LANGUAGE_GUESS.get(ext, '')
 
 
-def is_python2(files: List[str]) -> bool:
-    """Check if python2.
-
-    Args:
-        files (List[str]): Tuple of files.
-
-    Returns:
-       bool: True if python2, False otherwise.
-    """
-    python2 = re.compile(r'^\s*\bprint\b *[^ \(\),\]]|\braw_input\b')
-    for filename in files:
-        try:
-            with open(filename, 'r', encoding='utf-8') as f:
-                for index, line in enumerate(f):
-                    # print(index, line)
-                    if index == 0 and line.startswith('#!'):
-                        if 'python2' in line:
-                            return True
-                        if 'python3' in line:
-                            return False
-                    if python2.search(line.split('#')[0]):
-                        return True
-        except IOError:
-            return False
-    return False
-
-
 # flake8: noqa: C901
 def guess_mainfile(
         kat_language: str,

{kattis_cli-1.0.1.dist-info → kattis_cli-1.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kattis-cli
-Version: 1.0.1
+Version: 1.0.2
 Summary: A command-line tool for Kattis
 Home-page: https://github.com/rambasnet/kattis-cli
 Author: Ram Basnet
@@ -27,11 +27,16 @@ Description-Content-Type: text/markdown
 # Kattis-CLI
 
 Kattis CLI - download, test and submit Kattis problems using CLI.
-Based on Official Kattis CLI: [https://github.com/Kattis/kattis-cli](https://github.com/Kattis/kattis-cli)
+Inspired by the official Kattis CLI: [https://github.com/Kattis/kattis-cli](https://github.com/Kattis/kattis-cli)
+
+![Tests](https://github.com/rambasnet/kattis-cli/actions/workflows/ci-test.yml/badge.svg)
+[![PyPI version](https://badge.fury.io/py/kattis-cli.svg)](https://badge.fury.io/py/kattis-cli)
+[![PyPI - Downloads](https://img.shields.io/pypi/dm/kattis-cli)](https://pypi.org/project/kattis-cli/)
+[![PyPI - License](https://img.shields.io/pypi/l/kattis-cli)](https://pypi.org/project/kattis-cli/)
 
 ## Requirements
 
-- Python 3.8+ (PyPy preferred as Kattis uses PyPy to run your code)
+- Python 3.8+ (PyPy preferred as Kattis uses PyPy to run your Python3 solutions)
 - [Kattis account](https://open.kattis.com/login/email)
 
 ## Windows

{kattis_cli-1.0.1.dist-info → kattis_cli-1.0.2.dist-info}/RECORD

@@ -3,16 +3,16 @@ kattis_cli/__init__.py,sha256=afN92pog2fGyicY6KNBofYbCBYbYj4Fpi_INUSpsc-E,402
 kattis_cli/download.py,sha256=aa0hBg_6Ou8SGrufgkNUjfWEHdmiu7U-W4d-yWHZNOA,5877
 kattis_cli/kattis.py,sha256=t5fog-tx1f1GeM-D3aa2YhuhxTnai_kV-x7miZbs6SI,13767
 kattis_cli/kattis_setup.py,sha256=rq_-Fz8oafxFJ8MOcxJqHQT8-bezgVHZXY2hm4EQF3U,4357
-kattis_cli/main.py,sha256=PpizTV_9JDr-vgplpuwlgjJrxT62dMVLEC0zPm2R9Zo,3952
+kattis_cli/main.py,sha256=LDUUdG0eqiKL_zJX9V05OaqUIXKrOOwVAWOgN4TWWQQ,4121
 kattis_cli/settings.py,sha256=d5q4dYj9VqDSqPalleh2oZWtND-1bPB0T2IwdajFrBg,591
-kattis_cli/test_solution.py,sha256=M_BpwQC6lhWjk-d2oum4i5a_T25tAFgjWot9JFmH6R8,5659
+kattis_cli/test_solution.py,sha256=KKYedt3ISQYLIoahLPgPizpApLQnOrBhmkew5aHW9BM,6627
 kattis_cli/ui.py,sha256=dI06yncjm8sdLQkupPxGRODvQ6Gbawwuqn-67FrWg6I,2682
 kattis_cli/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kattis_cli/utils/config.py,sha256=cr5ssEVWolDbqP8krts0YSK5vSLatKTveeHBRNsLRC8,2277
 kattis_cli/utils/run_program.py,sha256=NWQ6vtTeWgkaW75r91FIHGXR5cAbeu8yMb5hwzpYFsg,2613
-kattis_cli/utils/utility.py,sha256=_uyPR_64_mX5EJDvdGf9fvjaAKwZKXvEcgksALQvml8,11500
-kattis_cli-1.0.1.dist-info/LICENSE,sha256=JmBa4SEKBCDWEgiOZcISU4tUCpli6xSpVlSYgkBXSNQ,1067
-kattis_cli-1.0.1.dist-info/METADATA,sha256=zpdJWccqlQ1P-qVRKui7278rMZjwBPU5mNG6KWqmFh8,5551
-kattis_cli-1.0.1.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-kattis_cli-1.0.1.dist-info/entry_points.txt,sha256=kyzGN20VqUPR_H0J_jJUKT-10-cAMFLVegQ6C7tbHss,47
-kattis_cli-1.0.1.dist-info/RECORD,,
+kattis_cli/utils/utility.py,sha256=JkypTfPOlDSxmZsh7KKcAVG2BC58t66corMBtcdvU7E,10532
+kattis_cli-1.0.2.dist-info/LICENSE,sha256=JmBa4SEKBCDWEgiOZcISU4tUCpli6xSpVlSYgkBXSNQ,1067
+kattis_cli-1.0.2.dist-info/METADATA,sha256=TImOazyOs7tJWBf_E-NP9x3-_InVey2bnqBxhjJMGOs,5961
+kattis_cli-1.0.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+kattis_cli-1.0.2.dist-info/entry_points.txt,sha256=kyzGN20VqUPR_H0J_jJUKT-10-cAMFLVegQ6C7tbHss,47
+kattis_cli-1.0.2.dist-info/RECORD,,

{kattis_cli-1.0.1.dist-info → kattis_cli-1.0.2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.8.1
+Generator: poetry-core 1.9.0
 Root-Is-Purelib: true
 Tag: py3-none-any