handle test class methods for the test diff

Author: mohammedahmed18

codeflash/models/models.py

@@ -515,6 +515,11 @@ def id(self) -> str:
             f"{self.function_getting_tested}:{self.iteration_id}"
         )
 
+    # TestSuiteClass.test_function_name
+    def test_fn_qualified_name(self) -> str:
+        class_prefix = f"{self.test_class_name}." if self.test_class_name else ""
+        return f"{class_prefix}{self.test_function_name}"
+
     def find_func_in_class(self, class_node: cst.ClassDef, func_name: str) -> Optional[cst.FunctionDef]:
         for stmt in class_node.body.body:
             if isinstance(stmt, cst.FunctionDef) and stmt.name.value == func_name:

codeflash/verification/equivalence.py

@@ -53,12 +53,14 @@ def compare_test_results(original_results: TestResults, candidate_results: TestR
         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_function_name, "")
+            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_function_name, "") if original_test_failures else ""
+            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:

tests/test_codeflash_capture.py

@@ -1406,7 +1406,6 @@ def risk_adjusted_return(return_val, weight):
         # Remove instrumentation
         FunctionOptimizer.write_code_and_helpers(candidate_fto_code, candidate_helper_code, fto.file_path)
         matched, diffs = compare_test_results(test_results, modified_test_results)
-        print(diffs)
 
         assert not matched