From 0cfde7d7b18867b850fa2b3bfe732bd5b8febf8a Mon Sep 17 00:00:00 2001 From: Bobby Wang Date: Fri, 18 Jun 2021 11:36:02 +0800 Subject: [PATCH] Enable window lead/lag test on struct --- java/src/test/java/ai/rapids/cudf/TableTest.java | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/java/src/test/java/ai/rapids/cudf/TableTest.java b/java/src/test/java/ai/rapids/cudf/TableTest.java index c94887d566d..3b0705aed3e 100644 --- a/java/src/test/java/ai/rapids/cudf/TableTest.java +++ b/java/src/test/java/ai/rapids/cudf/TableTest.java @@ -3321,9 +3321,7 @@ null, new StructData(13, "s13"), new StructData(14, "s14"), new StructData(-14, assertColumnsAreEqual(expectAggResult, windowAggResults.getColumn(0)); assertColumnsAreEqual(decExpectAggResult, decWindowAggResults.getColumn(0)); assertColumnsAreEqual(listExpectAggResult, listWindowAggResults.getColumn(0)); - - // TODO this is not gonna work, since libcudf has some issue for lead on struct with default values - // assertColumnsAreEqual(structExpectAggResult, structWindowAggResults.getColumn(0)); + assertColumnsAreEqual(structExpectAggResult, structWindowAggResults.getColumn(0)); } // Outside bounds @@ -3542,8 +3540,7 @@ null, new StructData(111, "s111"), new StructData(null, "s112"), new StructData( assertColumnsAreEqual(expectAggResult, windowAggResults.getColumn(0)); assertColumnsAreEqual(decExpectAggResult, decWindowAggResults.getColumn(0)); assertColumnsAreEqual(listExpectAggResult, listWindowAggResults.getColumn(0)); - // TODO this is not gonna work, since libcudf has some issue for lag on struct with default values - // assertColumnsAreEqual(structExpectAggResult, structWindowAggResults.getColumn(0)); + assertColumnsAreEqual(structExpectAggResult, structWindowAggResults.getColumn(0)); } // Outside bounds