testgenie-py 0.3.0__py3-none-any.whl → 0.3.2__py3-none-any.whl

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()

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):

testgen/controller/docker_controller.py

@@ -35,12 +35,22 @@ 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)
+            print(f"Copied file from {src_path} to {dest_path}")
+            self.logger.debug(f"Copied file from {src_path} to {dest_path}")
+        except Exception as e:
+            print(f"Failed to copy file: {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])}"
 

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
     

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.dist-info → testgenie_py-0.3.2.dist-info}/METADATA

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

{testgenie_py-0.3.0.dist-info → testgenie_py-0.3.2.dist-info}/RECORD

@@ -1,7 +1,6 @@
-testgen/.coverage,sha256=Cp0ihR5Imuuiso9j7j71iuvxJfMY6aeVWNHee--RrXA,69632
 testgen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 testgen/analyzer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-testgen/analyzer/ast_analyzer.py,sha256=fUtrJQsgnKeSKqKaXSO_ZsY_uQr2aPeyTQP9ToyFEg4,6886
+testgen/analyzer/ast_analyzer.py,sha256=y4ipJ59RfgJnhT_T1ZJP-Px3JINJovkPWMj7UpID3Ak,6824
 testgen/analyzer/contracts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 testgen/analyzer/contracts/contract.py,sha256=6rNYJOy_2GrOhGtaXTDIOX6pTEOqo856FxG0yTc8amI,369
 testgen/analyzer/contracts/no_exception_contract.py,sha256=rTWTuu5XKmvzBPD6yNAqiNehk9lbWn_Z8zFj-_djZ_w,512
@@ -11,21 +10,13 @@ testgen/analyzer/random_feedback_analyzer.py,sha256=hDcNIWE4BFtsoyXHWyxcC1qeq7y9
 testgen/analyzer/reinforcement_analyzer.py,sha256=U6W7g-xDFS_P0aqxjPRnaXjGp2gOC9TMdfWaetFV7eE,3568
 testgen/analyzer/test_case_analyzer.py,sha256=0qLcRKOrQ4Uyh1ZMecVTkY9bWjNnx7OridigNyjfUk4,1620
 testgen/analyzer/test_case_analyzer_context.py,sha256=7umDPHBjoTaCRvZOdNynpnWsr14Gy0E2hpMBPK8mz3Q,2128
-testgen/code_to_test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-testgen/code_to_test/boolean.py,sha256=RM2byJDKFTx1FFrxnPG_uZWb24OcOH20Uh-D1Z-yPcE,3373
-testgen/code_to_test/calculator.py,sha256=j7lw21BYg_vxaQVCrjOlhQOpObjSe5gu9xzthmdGphk,830
-testgen/code_to_test/code_to_fuzz.py,sha256=ZmASYCtvqfW-x70ynjy0o7iA9WLVbvLuf2n2K1fPE9Y,9188
-testgen/code_to_test/code_to_fuzz_lite.py,sha256=EjS4GB5klnJrvZIDDpS9iIt0iZETVJI0ez9DmihMiYc,10429
-testgen/code_to_test/decisions.py,sha256=ZiW-EF1yTOWeQVZ7Xh0oIPHK2NSNPUzVxCY0E2GbYvQ,1376
-testgen/code_to_test/math_utils.py,sha256=FqPQz5-e1oklzGyTVYNO8JMKDg5ob8z05VOlpKfKvvw,853
-testgen/code_to_test/sample_code_bin.py,sha256=kMvV_-oFeAkxUmDx41Pw0iTTmMG1pvX31TezXPWMOkE,4225
 testgen/controller/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-testgen/controller/cli_controller.py,sha256=A3TsJQT20UI2TlBqC7SyI8TG0n1vP_Itd1K8TRoD3ps,8735
-testgen/controller/docker_controller.py,sha256=M6Q_tiAGX8j4lH1NLUFMTCreKXmoKTEpgo60z-KWVRg,8343
+testgen/controller/cli_controller.py,sha256=DGPAt370TTMurL8gLKrVsl1yAP64a5OvgBBuBqhHX4I,9011
+testgen/controller/docker_controller.py,sha256=YRC-q6NIofwW5IQrgTU3Td2pz5P735thDNd1W3TGI0k,8720
 testgen/docker/Dockerfile,sha256=6mutzP0ZkVueuvMLCOy2bsyGxWjLHU-cq7RFkuC8r5Y,781
 testgen/generator/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 testgen/generator/code_generator.py,sha256=-Nh5Jt4my29YWA_LKWn98I4jvDA-fNi6AGuw-vJhRV0,3171
-testgen/generator/doctest_generator.py,sha256=Dkaf2l72ENk1mqNP3WkOnIY-iqrTHh-443AWtxzRWGY,8806
+testgen/generator/doctest_generator.py,sha256=B9I-nw7G7ubcx51XLOl1kiOEL0YW0aDuCAKebqfkRRI,8751
 testgen/generator/generator.py,sha256=9M9zg8DO0uU5W5s9hsz2oPCL7bz7sAD087M5TYottSs,2007
 testgen/generator/pytest_generator.py,sha256=ZjQKPHlizFK2yx9WabAxkW9sM-xa64d08IIQibnvwuk,3369
 testgen/generator/test_generator.py,sha256=D2Y3DaWH4fdIc5_9Xrznrkm0urFfhpYxuhL81M7RRaw,710
@@ -50,13 +41,11 @@ testgen/service/analysis_service.py,sha256=uPoPJFLh4Imf5y_Zd8pX9zxRijr-D0K0WV22Y
 testgen/service/cfg_service.py,sha256=Czr91D5AHVq38yPKyhZDjM5jMObGPvY0yPE-PxNVBYM,2326
 testgen/service/generator_service.py,sha256=3JCWRlg5UKNhKRG9VXIycDFmuYKsRTzEmhmtlxroWAc,8088
 testgen/service/logging_service.py,sha256=tCv9LHL1gUScQQp34_iYoUxfCFzLRtFG_DXRsJ73SjA,3189
-testgen/service/service.py,sha256=tZz6IDsR7FjK53G9In0Bq1RlQtp2SXgcslJJ7Tmhj4I,21220
+testgen/service/service.py,sha256=dhRvc-wzdn8uXozZTAx4qBVm7n48GOPue7AQ-ptMANk,22406
 testgen/sqlite/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 testgen/sqlite/db.py,sha256=RUt88ndT4CQRo9aZzLGXPHRK1iw0LyDEGnoWlOLzKGM,2390
 testgen/sqlite/db_service.py,sha256=F2ug_FBrYOQtJwM2cMkvQrTS54_VhWGDAHXndnTi-c8,9768
-testgen/testgen.db,sha256=Q70k6dPV651fhLhlYzN79KlCZD8zJexy7Cn7zGDIKHo,49152
-testgen/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-testgen/tests/test_code_to_fuzz.py,sha256=lShIQgyb1qIElcsbUZ8pLLJB6dY1KVAA_Q76Aks6Dvg,8696
+testgen/testgen.db,sha256=kXuNHy5rz2mmm9EQOgmhJDqHxbHZysBtc_8xlnMqxRw,53248
 testgen/tree/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 testgen/tree/node.py,sha256=ONJtbACShN4yNj1X-UslFRgLyBP6mrbg7qZr3c6dWyA,165
 testgen/tree/tree_utils.py,sha256=gT7jucky6_GWVOlDI6jpv6RMeWCkntGOHIYLvHxD85k,2122
@@ -72,7 +61,7 @@ testgen/util/z3_utils/branch_condition.py,sha256=N9FNR-iJmxIC62NpDQNVZ1OP14rXXqY
 testgen/util/z3_utils/constraint_extractor.py,sha256=RXJLpmk6dAvHZ27839VXKXNtdy9St1F-17-pSEFu4bM,1285
 testgen/util/z3_utils/variable_finder.py,sha256=dUh3F9_L_BDMz1ybiGss09LLcM_egbitgj0FT5Nh9u4,245
 testgen/util/z3_utils/z3_test_case.py,sha256=yF4oJOrXMLzOwDUqXdoeg83MOTl3pvc_lYaZcS01CuQ,4983
-testgenie_py-0.3.0.dist-info/METADATA,sha256=_z3mECZ_M5D3bTWvzG3zE3Qow3GnUP9yyqEIrhaax70,4350
-testgenie_py-0.3.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-testgenie_py-0.3.0.dist-info/entry_points.txt,sha256=OUN4GqB4zHlHWwWGjwIPbur4E_ZqQgkeeqaCLhzRZgg,47
-testgenie_py-0.3.0.dist-info/RECORD,,
+testgenie_py-0.3.2.dist-info/METADATA,sha256=m1oMD-i_FbjvMwQMpuQs-G7ID6SOv8ZxsK8OAUyS4Ew,4350
+testgenie_py-0.3.2.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+testgenie_py-0.3.2.dist-info/entry_points.txt,sha256=OUN4GqB4zHlHWwWGjwIPbur4E_ZqQgkeeqaCLhzRZgg,47
+testgenie_py-0.3.2.dist-info/RECORD,,

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"""

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'