Update hypothesis_testing.py

Author: KRRT7

codeflash/verification/hypothesis_testing.py

@@ -173,9 +173,9 @@ def generate_hypothesis_tests(
 
             class TestFunctionRemover(ast.NodeTransformer):
                 def visit_FunctionDef(self, node):  # noqa: ANN001, ANN202
-                    if function_to_optimize.function_name not in node.name:
-                        return None
-                    return node
+                    if node.name.startswith("test_") and function_to_optimize.function_name in node.name:
+                        return node
+                    return None
 
             modified_tree = TestFunctionRemover().visit(tree)
             ast.fix_missing_locations(modified_tree)