Function apply_test_changes

scripts/update_lib/cmd_auto_mark.py:589–617

Apply test changes to content. Args: contents: File content failing_tests: Set of (class_name, method_name) to mark as expectedFailure unexpected_successes: Set of (class_name, method_name) to remove expectedFailure error_messages: Dict mapping (class_name,

(
    contents: str,
    failing_tests: set[tuple[str, str]],
    unexpected_successes: set[tuple[str, str]],
    error_messages: dict[tuple[str, str], str] | None = None,
)

def apply_test_changes(
    contents: str,
    failing_tests: set[tuple[str, str]],
    unexpected_successes: set[tuple[str, str]],
    error_messages: dict[tuple[str, str], str] | None = None,
) -> str:
    """
    Apply test changes to content.

    Args:
        contents: File content
        failing_tests: Set of (class_name, method_name) to mark as expectedFailure
        unexpected_successes: Set of (class_name, method_name) to remove expectedFailure
        error_messages: Dict mapping (class_name, method_name) to error message

    Returns:
        Modified content
    """
    if unexpected_successes:
        contents = remove_expected_failures(contents, unexpected_successes)

    if failing_tests:
        failing_tests, error_messages = _consolidate_to_parent(
            contents, failing_tests, error_messages
        )
        patches = build_patches(failing_tests, error_messages)
        contents = apply_patches(contents, patches)

    return contents


def strip_reasonless_expected_failures(