Merge pull request #196 from codeflash-ai/cf-capture-fix

Codeflash Capture should not depend on anything inside codeflash

Author: aseembits93

codeflash/models/models.py

@@ -364,7 +364,7 @@ class TestingMode(enum.Enum):
     PERFORMANCE = "performance"
     LINE_PROFILE = "line_profile"
 
-
+#TODO this class is duplicated in codeflash_capture
 class VerificationType(str, Enum):
     FUNCTION_CALL = (
         "function_call"  # Correctness verification for a test function, checks input values and output values)

codeflash/verification/codeflash_capture.py

@@ -1,16 +1,22 @@
 from __future__ import annotations
 
+# This file should not have any dependencies on codeflash
 import functools
 import gc
 import inspect
 import os
 import sqlite3
 import time
 from pathlib import Path
-
+from enum import Enum
 import dill as pickle
 
-from codeflash.models.models import VerificationType
+class VerificationType(str, Enum):
+    FUNCTION_CALL = (
+        "function_call"  # Correctness verification for a test function, checks input values and output values)
+    )
+    INIT_STATE_FTO = "init_state_fto"  # Correctness verification for fto class instance attributes after init
+    INIT_STATE_HELPER = "init_state_helper"  # Correctness verification for helper class instance attributes after init
 
 
 def get_test_info_from_stack(tests_root: str) -> tuple[str, str | None, str, str]: