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

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

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

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

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

{testgenie_py-0.2.9 → 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.2.9 → 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.2.9 → 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.2.9 → 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.2.9 → 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
@@ -155,7 +157,15 @@ class Service:
         
         try:
             if self.test_format == UNITTEST_FORMAT:
-                self.execute_and_store_unittest(file_path_to_use, test_file)
+                 subprocess.run(["python", "-m", "coverage", "run", "--source=.", "-m", "unittest", test_file], check=True)
+                 result = subprocess.run(
+                     ["python", "-m", "coverage", "report", file_path_to_use], 
+                     check=True, 
+                     capture_output=True, 
+                     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:
@@ -305,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.2.9/testgen/generated_boolean.py

@@ -1,48 +0,0 @@
-def bin_and(a: bool, b: bool):
-    if a == True:
-        if b == True:
-            return True
-        else:
-            return False
-    else:
-        if b == True:
-            return False
-        else:
-            return False
-
-def bin_xor(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 status_flags(active: bool, verified: bool, admin: bool):
-    if active == True:
-        if verified == True:
-            if admin == True:
-                return 'admin-verified'
-            else:
-                return 'user-verified'
-        else:
-            if admin == True:
-                return 'admin-unverified'
-            else:
-                return 'user-unverified'
-    else:
-        if verified == True:
-            if admin == True:
-                return 'admin-verified'
-            else:
-                return 'inactive'
-        else:
-            if admin == True:
-                return 'admin-unverified'
-            else:
-                return 'inactive'
-

testgenie_py-0.2.9/testgen/generated_samplecodebin.py

@@ -1,545 +0,0 @@
-class GeneratedSampleCodeBin:
-    def bin_and(self, a: bool, b: bool):
-        """
-        >>> GeneratedSampleCodeBin().bin_and(True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().bin_and(True, False)
-        False
-
-        >>> GeneratedSampleCodeBin().bin_and(False, True)
-        False
-        """
-        if a == True:
-            if b == True:
-                return True
-            else:
-                return False
-        else:
-            if b == True:
-                return False
-            else:
-                return False
-    
-
-    def bin_and_bad(self, a: bool, b: bool):
-        """
-        >>> GeneratedSampleCodeBin().bin_and_bad(True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().bin_and_bad(True, False)
-        True
-
-        >>> GeneratedSampleCodeBin().bin_and_bad(False, True)
-        True
-        """
-        if a == True:
-            if b == True:
-                return True
-            else:
-                return True
-        else:
-            if b == True:
-                return True
-            else:
-                return False
-    
-
-    def bin_nand(self, a: bool, b: bool):
-        """
-        >>> GeneratedSampleCodeBin().bin_nand(True, True)
-        False
-
-        >>> GeneratedSampleCodeBin().bin_nand(True, False)
-        True
-
-        >>> GeneratedSampleCodeBin().bin_nand(False, True)
-        True
-        """
-        if a == True:
-            if b == True:
-                return False
-            else:
-                return True
-        else:
-            if b == True:
-                return True
-            else:
-                return True
-    
-
-    def bin_nor(self, a: bool, b: bool):
-        """
-        >>> GeneratedSampleCodeBin().bin_nor(True, True)
-        False
-
-        >>> GeneratedSampleCodeBin().bin_nor(True, False)
-        False
-
-        >>> GeneratedSampleCodeBin().bin_nor(False, True)
-        False
-        """
-        if a == True:
-            if b == True:
-                return False
-            else:
-                return False
-        else:
-            if b == True:
-                return False
-            else:
-                return True
-    
-
-    def bin_or(self, a: bool, b: bool):
-        """
-        >>> GeneratedSampleCodeBin().bin_or(True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().bin_or(True, False)
-        True
-
-        >>> GeneratedSampleCodeBin().bin_or(False, True)
-        True
-        """
-        if a == True:
-            if b == True:
-                return True
-            else:
-                return True
-        else:
-            if b == True:
-                return True
-            else:
-                return False
-    
-
-    def bin_or_bad(self, a: bool, b: bool):
-        """
-        >>> GeneratedSampleCodeBin().bin_or_bad(True, True)
-        False
-
-        >>> GeneratedSampleCodeBin().bin_or_bad(True, False)
-        False
-
-        >>> GeneratedSampleCodeBin().bin_or_bad(False, True)
-        False
-        """
-        if a == True:
-            if b == True:
-                return False
-            else:
-                return False
-        else:
-            if b == True:
-                return False
-            else:
-                return True
-    
-
-    def bin_xor(self, a: bool, b: bool):
-        """
-        >>> GeneratedSampleCodeBin().bin_xor(True, True)
-        False
-
-        >>> GeneratedSampleCodeBin().bin_xor(True, False)
-        True
-
-        >>> GeneratedSampleCodeBin().bin_xor(False, True)
-        True
-        """
-        if a == True:
-            if b == True:
-                return False
-            else:
-                return True
-        else:
-            if b == True:
-                return True
-            else:
-                return False
-    
-
-    def mux(self, c1: bool, c2: bool, x0: bool, x1: bool, x2: bool, x3: bool):
-        """
-        >>> GeneratedSampleCodeBin().mux(True, True, True, True, True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, True, True, True, True, False)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, True, True, True, False, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, True, True, True, False, False)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, True, True, False, True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, True, True, False, True, False)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, True, True, False, False, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, True, True, False, False, False)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, True, False, True, True, True)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, True, False, True, True, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, True, False, True, False, True)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, True, False, True, False, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, True, False, False, True, True)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, True, False, False, True, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, True, False, False, False, True)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, True, False, False, False, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, False, True, True, True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, False, True, True, True, False)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, False, True, True, False, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, False, True, True, False, False)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, False, True, False, True, True)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, False, True, False, True, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, False, True, False, False, True)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, False, True, False, False, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, False, False, True, True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, False, False, True, True, False)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, False, False, True, False, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, False, False, True, False, False)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(True, False, False, False, True, True)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, False, False, False, True, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, False, False, False, False, True)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(True, False, False, False, False, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, True, True, True, True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, True, True, True, True, False)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, True, True, True, False, True)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, True, True, True, False, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, True, True, False, True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, True, True, False, True, False)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, True, True, False, False, True)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, True, True, False, False, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, True, False, True, True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, True, False, True, True, False)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, True, False, True, False, True)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, True, False, True, False, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, True, False, False, True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, True, False, False, True, False)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, True, False, False, False, True)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, True, False, False, False, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, False, True, True, True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, False, True, True, True, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, False, True, True, False, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, False, True, True, False, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, False, True, False, True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, False, True, False, True, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, False, True, False, False, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, False, True, False, False, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, False, False, True, True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, False, False, True, True, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, False, False, True, False, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, False, False, True, False, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, False, False, False, True, True)
-        True
-
-        >>> GeneratedSampleCodeBin().mux(False, False, False, False, True, False)
-        False
-
-        >>> GeneratedSampleCodeBin().mux(False, False, False, False, False, True)
-        True
-        """
-        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
-    
-