Update test_unit_test_discovery.py

Author: KRRT7

tests/test_unit_test_discovery.py

@@ -1079,13 +1079,10 @@ def test_other():
         mock_function.function_name = "target_function"
         mock_function.parents = []  # No parent classes
 
-        filtered_tests = discover_unit_tests(test_config, functions_to_optimize=[mock_function])
-        # The import filter is designed for high recall, so it may include both functions
-        # because both test files import from the same module (mycode) that contains target_function
-        assert len(filtered_tests) >= 1  # Should find at least target_function
+        dummy_file_path = Path("dummy.py")
+        filtered_tests = discover_unit_tests(test_config, file_to_funcs_to_optimize={dummy_file_path: [mock_function]})
+        assert len(filtered_tests) >= 1
         assert "mycode.target_function" in filtered_tests
-        # In a perfect world we'd filter out other_function, but conservative filtering
-        # is acceptable for performance optimization purposes
 
 
 def test_analyze_imports_conditional_import():