Merge pull request #40 from codeflash-ai/cf551

[CF-551] Fix extra test type in TestType

Author: misrasaurabh1

codeflash/verification/test_results.py

@@ -33,7 +33,7 @@ class TestType(Enum):
     INIT_STATE_TEST = 6
 
     def to_name(self) -> str:
-        if self == TestType.INIT_STATE_TEST:
+        if self is TestType.INIT_STATE_TEST:
             return ""
         names = {
             TestType.EXISTING_UNIT_TEST: "⚙️ Existing Unit Tests",
@@ -166,6 +166,8 @@ def report_to_string(report: dict[TestType, dict[str, int]]) -> str:
     def report_to_tree(report: dict[TestType, dict[str, int]], title: str) -> Tree:
         tree = Tree(title)
         for test_type in TestType:
+            if test_type is TestType.INIT_STATE_TEST:
+                continue
             tree.add(
                 f"{test_type.to_name()} - Passed: {report[test_type]['passed']}, Failed: {report[test_type]['failed']}"
             )