Enable window lead/lag test on struct

rapidsai · Jun 18, 2021 · 0cfde7d · 0cfde7d
1 parent d8a3e60
commit 0cfde7d
Showing 1 changed file with 2 additions and 5 deletions.
diff --git 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