Function rows_same

python/ray/data/_internal/util.py:1886–1904

Check if two DataFrames have the same rows. Unlike the built-in pandas equals method, this function ignores indices and the order of rows. This is useful for testing Ray Data because its interface doesn't usually guarantee the order of rows.

(actual: pd.DataFrame, expected: pd.DataFrame)

def rows_same(actual: pd.DataFrame, expected: pd.DataFrame) -> bool:
    """Check if two DataFrames have the same rows.

    Unlike the built-in pandas equals method, this function ignores indices and the
    order of rows. This is useful for testing Ray Data because its interface doesn't
    usually guarantee the order of rows.
    """
    if len(actual) != len(expected):
        return False

    if len(actual) == 0:
        return True

    pd.testing.assert_frame_equal(
        _sort_df(actual).reset_index(drop=True),
        _sort_df(expected).reset_index(drop=True),
        check_dtype=False,
    )
    return True


def merge_resources_to_ray_remote_args(