kattis-cli 1.0.1__tar.gz → 1.0.2__tar.gz

{kattis_cli-1.0.1 → kattis_cli-1.0.2}/PKG-INFO

@@ -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 → kattis_cli-1.0.2}/README.md

@@ -1,11 +1,16 @@
 # 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 → kattis_cli-1.0.2}/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-1.0.1 → kattis_cli-1.0.2}/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-1.0.1 → kattis_cli-1.0.2}/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 → kattis_cli-1.0.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "kattis-cli"
-version = "1.0.1"
+version = "1.0.2"
 authors = ["Ram Basnet <rambasnet@gmail.com>"]
 description = "A command-line tool for Kattis"
 readme = "README.md"