testgenie-py 0.3.0__tar.gz → 0.3.1__tar.gz

{testgenie_py-0.3.0 → testgenie_py-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: testgenie-py
-Version: 0.3.0
+Version: 0.3.1
 Summary: Automated unit test generation tool for Python.
 Author: cjseitz
 Author-email: charlesjseitz@gmail.com

{testgenie_py-0.3.0 → testgenie_py-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "testgenie-py"
-version = "0.3.0"
+version = "0.3.1"
 description = "Automated unit test generation tool for Python."
 authors = ["cjseitz <charlesjseitz@gmail.com>"]
 readme = "README.md"

{testgenie_py-0.3.0 → testgenie_py-0.3.1}/testgen/analyzer/ast_analyzer.py

@@ -56,7 +56,6 @@ class ASTAnalyzer(TestCaseAnalyzerStrategy, ABC):
                         generated_file = function_metadata.filename
                         module = testgen.util.file_utils.load_module(generated_file)
                         
-                        # Check if we're dealing with a class
                         if function_metadata.class_name:
                             cls = getattr(module, function_metadata.class_name)
                             instance = cls()

{testgenie_py-0.3.0 → testgenie_py-0.3.1}/testgen/controller/cli_controller.py

@@ -46,6 +46,10 @@ class CLIController:
             print(f"Querying database for file: {args.file_path}")
             self.service.query_test_file_data(args.file_path)
             return
+        
+        if args.coverage:
+            self.service.get_coverage(args.file_path)
+            return
             
         running_in_docker = os.environ.get("RUNNING_IN_DOCKER") is not None
         if running_in_docker:
@@ -159,6 +163,11 @@ class CLIController:
             type=str,
             help="Path to log file (if not specified, logs will only go to console)"
         )
+        parser.add_argument(
+            "-c", "--coverage",
+            action="store_true",
+            help="Run coverage analysis on the generated tests"
+        )
         return parser
 
     def set_test_format(self, args: argparse.Namespace):

{testgenie_py-0.3.0 → testgenie_py-0.3.1}/testgen/controller/docker_controller.py

@@ -35,12 +35,20 @@ class DockerController:
 
         # Check if Docker image exists, build it if not
         image_name = "testgen-runner"
-        # If args.safe is set to false it means the image was not found and the system will try to run_locally
         self.get_image(docker_client, image_name, project_root)
         if not self.args.safe:
             self.logger.info("Docker image not found. Running locally...")
             return False
 
+        src_path = os.path.abspath(args.file_path)
+        dest_path = os.path.join(project_root, os.path.basename(src_path))
+        try:
+            shutil.copy(src_path, dest_path)
+            self.logger.debug(f"Copied file from {src_path} to {dest_path}")
+        except Exception as e:
+            self.logger.error(f"Failed to copy file: {e}")
+            sys.exit(1)
+
         docker_args = [args.file_path] + [arg for arg in sys.argv[2:] if arg != "--safe"]
         docker_args[0] = f"/controller/testgen/code_to_test/{os.path.basename(docker_args[0])}"
 

{testgenie_py-0.3.0 → testgenie_py-0.3.1}/testgen/generator/doctest_generator.py

@@ -9,7 +9,6 @@ class DocTestGenerator(TestGenerator):
     def __init__(self, generator_context: GeneratorContext):
         super().__init__(generator_context)
 
-    # Implementing abstract methods from TestGenerator
     def generate_test_header(self):
         pass
     

{testgenie_py-0.3.0 → testgenie_py-0.3.1}/testgen/service/service.py

@@ -8,6 +8,8 @@ import sqlite3
 import sys
 import time
 import subprocess
+
+import coverage
 import testgen.util.coverage_utils as coverage_utils
 from types import ModuleType
 from typing import List
@@ -163,6 +165,7 @@ class Service:
                      text=True
                  )
                  coverage_output = result.stdout
+                 print(coverage_output)
             elif self.test_format == PYTEST_FORMAT:
                 self.execute_and_store_pytest(test_file)
             elif self.test_format == DOCTEST_FORMAT:
@@ -312,6 +315,36 @@ class Service:
         """Create an analysis context for the given file."""
         return self.analysis_service.create_analysis_context(filepath)
 
+    def get_coverage(self, file_path: str):
+        """
+        Use the coverage library to calculate and print the coverage for the specified Python file.
+        Dynamically determine the source directory based on the file being tested.
+        """
+        # Dynamically determine the source directory
+        source_dir = os.path.dirname(file_path)
+        cov = coverage.Coverage(source=[source_dir])  # Use the directory of the file as the source
+        cov.start()
+
+        try:
+            # Dynamically import and execute the specified file
+            file_name = os.path.basename(file_path)
+            module_name = file_name.rstrip(".py")
+            spec = importlib.util.spec_from_file_location(module_name, file_path)
+            module = importlib.util.module_from_spec(spec)
+            spec.loader.exec_module(module)
+
+        except Exception as e:
+            print(f"Error while executing the file: {e}")
+            return
+
+        finally:
+            cov.stop()
+            cov.save()
+
+        # Report the coverage
+        print(f"Coverage report for {file_path}:")
+        cov.report(file=sys.stdout)
+
     @staticmethod
     def wait_for_file(file_path, retries=5, delay=1):
         """Wait for the generated file to appear."""

testgenie_py-0.3.0/testgen/code_to_test/boolean.py

@@ -1,146 +0,0 @@
-from typing import List
-
-
-def bin_and(a: bool, b: bool) ->bool:
-    """
-    >>> bin_and(True, False)
-    False
-
-    >>> bin_and(False, True)
-    False
-
-    >>> bin_and(False, False)
-    False
-
-    >>> bin_and(True, True)
-    True
-    """
-    if a == True:
-        if b == True:
-            return True
-        else:
-            return False
-    else:
-        return False
-
-
-def bin_xor(a: bool, b: bool) ->bool:
-    """
-    >>> bin_xor(True, True)
-    False
-
-    >>> bin_xor(True, False)
-    True
-
-    >>> bin_xor(False, False)
-    False
-
-    >>> bin_xor(False, True)
-    True
-    """
-    if a == True:
-        if b == True:
-            return False
-        else:
-            return True
-    elif b == True:
-        return True
-    else:
-        return False
-
-
-def status_flags(active: bool, verified: bool, admin: bool) ->str:
-    """
-    >>> status_flags(False, False, False)
-    'inactive'
-
-    >>> status_flags(False, False, True)
-    'admin-unverified'
-
-    >>> status_flags(True, True, False)
-    'user-verified'
-
-    >>> status_flags(True, True, True)
-    'admin-verified'
-
-    >>> status_flags(True, False, False)
-    'user-unverified'
-
-    >>> status_flags(False, True, True)
-    'admin-verified'
-
-    >>> status_flags(False, True, False)
-    'inactive'
-
-    >>> status_flags(True, False, True)
-    'admin-unverified'
-    """
-    if admin:
-        if verified:
-            return 'admin-verified'
-        else:
-            return 'admin-unverified'
-    elif active:
-        if verified:
-            return 'user-verified'
-        else:
-            return 'user-unverified'
-    else:
-        return 'inactive'
-
-
-"""def half_adder(a: bool, b: bool) ->tuple:
-    sum: bool = bin_xor(a, b)
-    carry: bool = bin_and(a, b)
-    return sum, carry
-
-
-def full_adder(a: bool, b: bool, carry_in: bool) ->tuple:
-    sum1, carry = half_adder(a, b)
-    sum2, carry_out = half_adder(sum1, carry_in)
-    return sum2, carry or carry_out
-
-
-def thirty_two_bit_adder_excep(x: int, y: int) ->List[int]:
-    x_bits: List[int] = bit_converter(x)
-    y_bits: List[int] = bit_converter(y)
-    result: List[int] = [0] * 32
-    carry: bool = False
-    for i in range(32):
-        try:
-            sum_bit, carry = full_adder(x_bits[i], y_bits[i], carry)
-            result[i] = sum_bit
-        except IndexError as e:
-            print(f'Index Out of Bounds Error In ThirtyTwoBitAdder: {e}')
-            result = [1] * 32
-        if carry:
-            return OverflowError('Sum exceeds 32 bits')
-    return result
-
-
-def thirty_two_bit_adder(x: int, y: int) ->List[int]:
-    print(x.bit_length() - 1)
-    print(y.bit_length() - 1)
-    x_bits: List[int] = bit_converter(x)
-    y_bits: List[int] = bit_converter(y)
-    result: List[int] = [0] * 32
-    carry: bool = False
-    for i in range(32):
-        sum_bit, carry = full_adder(x_bits[i], y_bits[i], carry)
-        result[i] = sum_bit
-    return result
-
-def bit_converter(num: int) ->List[int]:
-    binary_str: str = bin(num)[2:]
-    print(bin(num)[2:])
-    return [int(digit) for digit in binary_str]
-
-def thirty_two_bit_adder_excep(x: int, y: int) ->List[int]:
-    x_bits: List[int] = bit_converter(x)
-    y_bits: List[int] = bit_converter(y)
-    result: List[int] = [0] * 32
-    carry: bool = False
-    for i in range(32):
-        sum_bit, carry = full_adder(x_bits[i], y_bits[i], carry)
-        result[i] = sum_bit
-    return result"""

testgenie_py-0.3.0/testgen/code_to_test/calculator.py

@@ -1,29 +0,0 @@
-class AdvancedCalculator:
-
-    def evaluate(self, a: int, b: int, op: str) ->(int | float | str):
-        """>>> AdvancedCalculator().evaluate(16, 56, 'abc')
-'invalid'"""
-        if op == 'add':
-            return a + b
-        elif op == 'sub':
-            return a - b
-        elif op == 'mul':
-            return a * b
-        elif op == 'div':
-            if b == 0:
-                return 'undefined'
-            return a / b
-        else:
-            return 'invalid'
-
-    def is_in_range(self, val: float) ->str:
-        """>>> AdvancedCalculator().is_in_range(0.4023417704905361)
-'fractional'"""
-        if val < 0.0:
-            return 'below zero'
-        elif 0.0 <= val < 1.0:
-            return 'fractional'
-        elif 1.0 <= val <= 100.0:
-            return 'valid'
-        else:
-            return 'out of range'

testgenie_py-0.3.0/testgen/code_to_test/code_to_fuzz.py

@@ -1,329 +0,0 @@
-def random_fuzz_code(x: int, y: int) -> int:
-    result = x
-    if x < y:
-        result += y
-    else:
-        result -= y
-    return result
-
-
-def bin_and(a: bool, b: bool) -> bool:
-    if a == True:
-        if b == True:
-            return True
-        else:
-            return False
-    else:
-        return False
-
-
-def pos_or_neg(i: int):
-    if i > 0:
-        sgn = 1
-    else:
-        sgn = -1
-    if sgn > 0:
-        return 1
-    elif sgn < 0:
-        return -1
-    # else sgn == 0
-    # no return here
-
-
-def int_even(x: int):
-    if x % 2 == 0:
-        return True
-    else:
-        return False
-
-def http_code(code: int) ->str:
-    if code < 100 or code > 599:
-        return 'invalid'
-    elif 100 <= code < 200:
-        return 'informational'
-    elif 200 <= code < 300:
-        return 'success'
-    elif 300 <= code < 400:
-        return 'redirection'
-    elif 400 <= code < 500:
-        return 'client error'
-    else:
-        return 'server error'
-    
-def add_or_subtract(x: int, y: int) ->int:
-    result = x
-    if x < y:
-        result += y
-    else:
-        result -= y
-    return result
-    
-def bin_and_bad_generated_function(a: bool, b: bool):
-    if a == True:
-        if b == True:
-            return True
-        else:
-            return True
-    else:
-        if b == True:
-            return True
-        else:
-            return False
-
-
-def bin_nand_generated_function(a: bool, b: bool):
-    if a == True:
-        if b == True:
-            return False
-        else:
-            return True
-    else:
-        if b == True:
-            return True
-        else:
-            return True
-
-
-def bin_nor_generated_function(a: bool, b: bool):
-    if a == True:
-        if b == True:
-            return False
-        else:
-            return False
-    else:
-        if b == True:
-            return False
-        else:
-            return True
-
-
-def bin_or_generated_function(a: bool, b: bool):
-    if a == True:
-        if b == True:
-            return True
-        else:
-            return True
-    else:
-        if b == True:
-            return True
-        else:
-            return False
-
-
-def bin_or_bad_generated_function(a: bool, b: bool):
-    if a == True:
-        if b == True:
-            return False
-        else:
-            return False
-    else:
-        if b == True:
-            return False
-        else:
-            return True
-
-
-def bin_xor_generated_function(a: bool, b: bool):
-    if a == True:
-        if b == True:
-            return False
-        else:
-            return True
-    else:
-        if b == True:
-            return True
-        else:
-            return False
-
-
-def mux_generated_function(c1: bool, c2: bool, x0: bool, x1: bool, x2: bool, x3: bool):
-    if c1 == True:
-        if c2 == True:
-            if x0 == True:
-                if x1 == True:
-                    if x2 == True:
-                        if x3 == True:
-                            return True
-                        else:
-                            return True
-                    else:
-                        if x3 == True:
-                            return True
-                        else:
-                            return True
-                else:
-                    if x2 == True:
-                        if x3 == True:
-                            return True
-                        else:
-                            return True
-                    else:
-                        if x3 == True:
-                            return True
-                        else:
-                            return True
-            else:
-                if x1 == True:
-                    if x2 == True:
-                        if x3 == True:
-                            return False
-                        else:
-                            return False
-                    else:
-                        if x3 == True:
-                            return False
-                        else:
-                            return False
-                else:
-                    if x2 == True:
-                        if x3 == True:
-                            return False
-                        else:
-                            return False
-                    else:
-                        if x3 == True:
-                            return False
-                        else:
-                            return False
-        else:
-            if x0 == True:
-                if x1 == True:
-                    if x2 == True:
-                        if x3 == True:
-                            return True
-                        else:
-                            return True
-                    else:
-                        if x3 == True:
-                            return True
-                        else:
-                            return True
-                else:
-                    if x2 == True:
-                        if x3 == True:
-                            return False
-                        else:
-                            return False
-                    else:
-                        if x3 == True:
-                            return False
-                        else:
-                            return False
-            else:
-                if x1 == True:
-                    if x2 == True:
-                        if x3 == True:
-                            return True
-                        else:
-                            return True
-                    else:
-                        if x3 == True:
-                            return True
-                        else:
-                            return True
-                else:
-                    if x2 == True:
-                        if x3 == True:
-                            return False
-                        else:
-                            return False
-                    else:
-                        if x3 == True:
-                            return False
-                        else:
-                            return False
-    else:
-        if c2 == True:
-            if x0 == True:
-                if x1 == True:
-                    if x2 == True:
-                        if x3 == True:
-                            return True
-                        else:
-                            return True
-                    else:
-                        if x3 == True:
-                            return False
-                        else:
-                            return False
-                else:
-                    if x2 == True:
-                        if x3 == True:
-                            return True
-                        else:
-                            return True
-                    else:
-                        if x3 == True:
-                            return False
-                        else:
-                            return False
-            else:
-                if x1 == True:
-                    if x2 == True:
-                        if x3 == True:
-                            return True
-                        else:
-                            return True
-                    else:
-                        if x3 == True:
-                            return False
-                        else:
-                            return False
-                else:
-                    if x2 == True:
-                        if x3 == True:
-                            return True
-                        else:
-                            return True
-                    else:
-                        if x3 == True:
-                            return False
-                        else:
-                            return False
-        else:
-            if x0 == True:
-                if x1 == True:
-                    if x2 == True:
-                        if x3 == True:
-                            return True
-                        else:
-                            return False
-                    else:
-                        if x3 == True:
-                            return True
-                        else:
-                            return False
-                else:
-                    if x2 == True:
-                        if x3 == True:
-                            return True
-                        else:
-                            return False
-                    else:
-                        if x3 == True:
-                            return True
-                        else:
-                            return False
-            else:
-                if x1 == True:
-                    if x2 == True:
-                        if x3 == True:
-                            return True
-                        else:
-                            return False
-                    else:
-                        if x3 == True:
-                            return True
-                        else:
-                            return False
-                else:
-                    if x2 == True:
-                        if x3 == True:
-                            return True
-                        else:
-                            return False
-                    else:
-                        if x3 == True:
-                            return True
-                        else:
-                            return False