Skip to content

Conversation

@ShreyeshArangath
Copy link
Contributor

Which issue does this PR close?

Closes #1814

Rationale for this change

This change is part of the effort for enhancing Auron's correctness testing (#1745).

What changes are included in this PR?

This change adds supports for sort operator related tests.

Are there any user-facing changes?

N/A

How was this patch tested?

Existing unit tests

@ShreyeshArangath ShreyeshArangath marked this pull request as ready for review January 15, 2026 06:00
@ShreyeshArangath
Copy link
Contributor Author

@richox Can we merge this in? Thanks!

@ShreyeshArangath
Copy link
Contributor Author

@cxzl25 Could you PTAL? Thanks!

Copy link
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

This PR adds comprehensive correctness tests for the Sort operator in Spark 3.3 as part of the broader Auron correctness testing effort (issue #1745). The tests verify sort-related functionality including range partitioning, local sorting within partitions, exchange reuse optimization, and edge cases for large expressions and view changes.

Changes:

  • Added 5 new test cases in AuronDataFrameSuite covering repartitionByRange, distributeBy/localSort, exchange reuse, large CaseWhen expressions, and view change scenarios
  • Configured test suite to exclude the original Spark DataFrameSuite tests that have Auron-specific implementations
  • Added exclusions for tests that fail due to native execution differences

Reviewed changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 1 comment.

File Description
auron-spark-tests/spark33/src/test/scala/org/apache/spark/sql/AuronDataFrameSuite.scala New test suite with 5 sort-related tests and helper methods for verifying aggregation and shuffle behavior
auron-spark-tests/spark33/src/test/scala/org/apache/auron/utils/AuronSparkTestSettings.scala Configured AuronDataFrameSuite with appropriate test exclusions for Auron-specific implementations and known failures

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

@cxzl25 cxzl25 merged commit b1b5fef into apache:master Jan 21, 2026
104 checks passed
@ShreyeshArangath
Copy link
Contributor Author

Thanks! @cxzl25

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[Correctness Testing] Spark 3.3 - Add Sort operator related tests

3 participants