testgenie-py 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

testgen/analyzer/ast_analyzer.py

@@ -8,6 +8,9 @@ from abc import ABC
 from testgen.models.function_metadata import FunctionMetadata
 
 class ASTAnalyzer(TestCaseAnalyzerStrategy, ABC):
+    def __init__(self, analysis_context=None):
+        super().__init__(analysis_context)
+
     def collect_test_cases(self, function_metadata: FunctionMetadata) -> List[TestCase]:
         """Collect test cases by analyzing AST conditions and return statements"""
 
@@ -34,7 +37,7 @@ class ASTAnalyzer(TestCaseAnalyzerStrategy, ABC):
         for node in func_node_body:
             if isinstance(node, ast.If):
                 condition_str = self.parse_condition(node.test)
-                print(f"Condition found in function: {condition_str}")
+                self.logger.debug(f"Condition found in function: {condition_str}")
                 self.get_conditions_recursively(function_metadata, func_name, node.body, param_names, test_cases, conditions + [condition_str])
                 if node.orelse:
                     self.get_conditions_recursively(function_metadata, func_name, node.orelse, param_names, test_cases, conditions)

testgen/analyzer/fuzz_analyzer.py

@@ -61,8 +61,8 @@ class FuzzAnalyzer(TestCaseAnalyzerStrategy, ABC):
                 covered_branches = self.get_branch_coverage(module)
                 covered_branches_tuple = tuple(covered_branches)
 
-                print(f"[COVERED_BRANCHES]: {covered_branches}")
-                print(f"[EXECUTED BRANCHES]: {self.executed_branches}")
+                self.logger.debug(f"[COVERED_BRANCHES]: {covered_branches}")
+                self.logger.debug(f"[EXECUTED BRANCHES]: {self.executed_branches}")
 
                 for branch in covered_branches:
                     if branch[1] < 0:

testgen/analyzer/random_feedback_analyzer.py

@@ -74,7 +74,7 @@ class RandomFeedbackAnalyzer(TestCaseAnalyzerStrategy, ABC):
                         break
                 else:
                     # Optionally log duplicate detection
-                    print(f"Skipping duplicate test case: {func_name}{test_case.inputs}")
+                    self.logger.debug(f"Skipping duplicate test case: {func_name}{test_case.inputs}")
                             
             except Exception as e:
                 print(f"Error testing {function_metadata.function_name}: {e}")
@@ -97,10 +97,10 @@ class RandomFeedbackAnalyzer(TestCaseAnalyzerStrategy, ABC):
                                                                          func.function_name, test_case.inputs)
             covered = coverage_utils.get_list_of_covered_statements(analysis)
             self.covered_lines[func.function_name].update(covered)
-            print(f"Covered lines for {func.function_name}: {self.covered_lines[func.function_name]}")
+            self.logger.debug(f"Covered lines for {func.function_name}: {self.covered_lines[func.function_name]}")
 
         executable_statements = set(self.get_all_executable_statements(func))
-        print(f"Executable statements for {func.function_name}: {executable_statements}")
+        self.logger.debug(f"Executable statements for {func.function_name}: {executable_statements}")
 
         return self.covered_lines[func.function_name] == executable_statements
 

testgen/analyzer/test_case_analyzer.py

@@ -5,10 +5,12 @@ from typing import List, Dict
 from testgen.models.test_case import TestCase
 from testgen.models.analysis_context import AnalysisContext
 from testgen.models.function_metadata import FunctionMetadata
+from testgen.service.logging_service import get_logger
 
 class TestCaseAnalyzerStrategy(ABC):
     def __init__(self, analysis_context: AnalysisContext = None):
         self._analysis_context = analysis_context
+        self.logger = get_logger()
 
     @abstractmethod
     def collect_test_cases(self, function_metadata: FunctionMetadata) -> List[TestCase]:

testgen/code_to_test/boolean.py

@@ -0,0 +1,146 @@
+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

@@ -0,0 +1,29 @@
+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'

testgen/code_to_test/code_to_fuzz.py

@@ -0,0 +1,234 @@
+from typing import List
+
+
+def bin_and(a: bool, b: bool) ->bool:
+    """"
+    >>> bin_and(True, True)
+    True
+    >>> bin_and(True, False)
+    False
+    >>> bin_and(False, True)
+    False
+    >>> bin_and(False, False)
+    False
+
+    Examples:
+    >>> bin_and(True, False)
+    False"""
+    if a == True:
+        if b == True:
+            return True
+        else:
+            return False
+    else:
+        return False
+
+
+def bin_or(a: bool, b: bool) ->bool:
+    """"
+>>> bin_or(True, True)
+True
+>>> bin_or(True, False)
+True
+>>> bin_or(False, True)
+True
+>>> bin_or(False, False)
+False
+
+Examples:
+>>> bin_or(False, True)
+True"""
+    return a | b
+
+
+def bin_xor(a: bool, b: bool) ->bool:
+    """>>> bin_xor(True, True)
+False
+>>> bin_xor(True, False)
+True
+>>> bin_xor(False, True)
+True
+>>> bin_xor(False, False)
+False
+
+Examples:
+>>> bin_xor(True, True)
+False"""
+    return a ^ b
+
+
+def bin_nand(a: bool, b: bool) ->bool:
+    """>>> bin_nand(True, True)
+False
+>>> bin_nand(True, False)
+True
+>>> bin_nand(False, True)
+True
+>>> bin_nand(False, False)
+True
+
+Examples:
+>>> bin_nand(True, False)
+True"""
+    return not (a and b)
+
+
+def bin_nor(a: bool, b: bool) ->bool:
+    """>>> bin_nor(True, True)
+False
+>>> bin_nor(True, False)
+False
+>>> bin_nor(False, True)
+False
+>>> bin_nor(False, False)
+True
+
+Examples:
+>>> bin_nor(False, True)
+False"""
+    return not a | b
+
+
+def mux(c1: bool, c2: bool, x0: bool, x1: bool, x2: bool, x3: bool) ->bool:
+    """"
+>>> mux(1, 1, True, False, False, False)
+True
+>>> mux(1, 0, False, True, False, False)
+True
+>>> mux(0, 1, False, False, True, False)
+True
+>>> mux(0, 0, False, False, False, True)
+True
+>>> mux(0, 0, False, False, False, False)
+False
+
+Examples:
+>>> mux(False, False, False, False, False, True)
+True"""
+    return (c1 and c2 and x0) | (c1 and not c2 and x1) | (not c1 and c2 and x2
+        ) | (not c1 and not c2 and x3)
+
+
+def bin_and_bad(a: bool, b: bool) ->bool:
+    """"
+>>> bin_and(True, True)
+True
+>>> bin_and(True, False)
+False
+>>> bin_and(False, True)
+False
+>>> bin_and(False, False)
+False
+
+Examples:
+>>> bin_and_bad(False, True)
+True"""
+    if a == True and b == True:
+        return a and b
+    return a or b
+
+
+def bin_or_bad(a: bool, b: bool) ->bool:
+    """"
+>>> bin_or(True, True)
+True
+>>> bin_or(True, False)
+True
+>>> bin_or(False, True)
+True
+>>> bin_or(False, False)
+False
+
+Examples:
+>>> bin_or_bad(True, True)
+False"""
+    return not a | b
+
+
+def bit_converter_excep(num: int) ->List[int]:
+    """>>> bit_converter_excep(30)
+[1, 1, 1, 1, 0]"""
+    bs = bin(num)
+    binary_str = bs[2:]
+    try:
+        return [int(digit) for digit in binary_str]
+    except Exception as e:
+        print(binary_str)
+        print(bs)
+        print(f'Invalid Literal Exception in Bit Converter: {e}')
+        return
+
+
+def bit_converter(num: int) ->List[int]:
+    """
+    >>> bit_converter(25)
+    [1, 1, 0, 0, 1]
+    """
+    binary_str: str = bin(num)[2:]
+    print(bin(num)[2:])
+    return [int(digit) for digit in binary_str]
+
+
+def half_adder(a: bool, b: bool) ->tuple:
+    """>>> half_adder(True, True)
+(False, True)"""
+    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:
+    """>>> full_adder(True, False, False)
+(True, False)"""
+    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]:
+    """>>> thirty_two_bit_adder_excep(9, 3)
+None
+
+Examples:
+>>> thirty_two_bit_adder_excep(34, 1)
+None"""
+    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]:
+    """>>> thirty_two_bit_adder(8, 45)
+None"""
+    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 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