-
Notifications
You must be signed in to change notification settings - Fork 915
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
add unit tests for lead/lag on list for row window [skip ci] #8259
Conversation
LGTM |
LGTM, but some tests have failed. |
@@ -3464,6 +3681,16 @@ void testRangeWindowingLead() { | |||
.column(1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1) // GBY Key | |||
.column(0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2, 2) // GBY Key | |||
.column(7, 5, 1, 9, 7, 9, 8, 2, 8, 0, 6, 6, 8) // Agg Column | |||
.column(new ListType(false, new BasicType(true, DType.INT32)), |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Not a review suggestion. Just FYI: I think SparkSQL doesn't support calling LEAD()
/LAG()
on RANGE
windows. It might be worth checking. This path might simply not get exercised from the spark-rapids
plugin.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Removed the lead/lag for range window test
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Some minor changes requested. The test results look good to my eye.
Signed-off-by: Bobby Wang <[email protected]>
@mythrocks Could you help to check it |
@gpucibot merge |
This PR just added the unit tests for the lead/lag on List/Struct type for both Row-window and Range-window, Since libcudf has already supported the functionality. See the plugin feature request NVIDIA/spark-rapids#2012
But there is an issue when setting the default values for lead/lag on List/Struct. I just commented the corresponding assertation out. I will re-enable it after libcudf fixes it.