make it work

Author: mohammedahmed18

codeflash/models/models.py

@@ -383,7 +383,7 @@ class TestsInFile:
     test_type: TestType
 
 
-class OptimizedCandidateSource(enum.Enum, str):
+class OptimizedCandidateSource(str, Enum):
     OPTIMIZE = "OPTIMIZE"
     OPTIMIZE_LP = "OPTIMIZE_LP"
     REFINE = "REFINE"

codeflash/verification/equivalence.py

@@ -28,18 +28,6 @@ def compare_test_results(original_results: TestResults, candidate_results: TestR
     for test_id in test_ids_superset:
         original_test_result = original_results.get_by_unique_invocation_loop_id(test_id)
         cdd_test_result = candidate_results.get_by_unique_invocation_loop_id(test_id)
-        candidate_test_failures = candidate_results.test_failures
-        original_test_failures = original_results.test_failures
-        cdd_pytest_error = (
-            candidate_test_failures.get(original_test_result.id.test_fn_qualified_name(), "")
-            if candidate_test_failures
-            else ""
-        )
-        original_pytest_error = (
-            original_test_failures.get(original_test_result.id.test_fn_qualified_name(), "")
-            if original_test_failures
-            else ""
-        )
 
         if cdd_test_result is not None and original_test_result is None:
             continue
@@ -51,7 +39,7 @@ def compare_test_results(original_results: TestResults, candidate_results: TestR
         ):
             continue
         if original_test_result is None or cdd_test_result is None:
-            return False, []
+            continue
         did_all_timeout = did_all_timeout and original_test_result.timed_out
         if original_test_result.timed_out:
             continue
@@ -62,6 +50,19 @@ def compare_test_results(original_results: TestResults, candidate_results: TestR
         ):
             superset_obj = True
 
+        candidate_test_failures = candidate_results.test_failures
+        original_test_failures = original_results.test_failures
+        cdd_pytest_error = (
+            candidate_test_failures.get(original_test_result.id.test_fn_qualified_name(), "")
+            if candidate_test_failures
+            else ""
+        )
+        original_pytest_error = (
+            original_test_failures.get(original_test_result.id.test_fn_qualified_name(), "")
+            if original_test_failures
+            else ""
+        )
+
         test_src_code = original_test_result.id.get_src_code(original_test_result.file_name)
         test_diff = TestDiff(
             scope=TestDiffScope.RETURN_VALUE,