Corrected CI test failure.

deephaven · Dec 18, 2024 · 030069d · 030069d
1 parent fd4fa25
commit 030069d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 11 deletions.
diff --git a/java-client/session-dagger/src/test/java/io/deephaven/client/AggBySessionTest.java b/java-client/session-dagger/src/test/java/io/deephaven/client/AggBySessionTest.java
@@ -4,16 +4,8 @@
 package io.deephaven.client;
 
 import io.deephaven.api.ColumnName;
-import io.deephaven.api.agg.Aggregation;
+import io.deephaven.api.agg.*;
 import io.deephaven.api.agg.Aggregation.Visitor;
-import io.deephaven.api.agg.Aggregations;
-import io.deephaven.api.agg.ColumnAggregation;
-import io.deephaven.api.agg.ColumnAggregations;
-import io.deephaven.api.agg.Count;
-import io.deephaven.api.agg.FirstRowKey;
-import io.deephaven.api.agg.Formula;
-import io.deephaven.api.agg.LastRowKey;
-import io.deephaven.api.agg.Partition;
 import io.deephaven.api.object.UnionObject;
 import io.deephaven.qst.table.TableSpec;
 import io.deephaven.qst.table.TimeTable;
@@ -168,6 +160,11 @@ public void visit(Count count) {
             out.add(Count.of("MyCount"));
         }
 
+        @Override
+        public void visit(CountWhere countWhwere) {
+            out.add(CountWhere.of("CountWhere", "B > 0", "I <= 5"));
+        }
+
         @Override
         public void visit(FirstRowKey firstRowKey) {
             out.add(FirstRowKey.of("First"));

diff --git a/py/client/tests/test_table.py b/py/client/tests/test_table.py
@@ -305,7 +305,6 @@ def test_agg_all_by(self):
             with self.assertRaises(DHError) as cm:
                 test_table.agg_all_by(agg=count_(col="ca"), by=["a"])
 
-
     def test_agg_count_where_output(self):
         """
         Test and validation of the agg_count_where feature
@@ -326,7 +325,6 @@ def test_agg_count_where_output(self):
         self.assertEqual(df.loc[0, "count2"], 17)
         self.assertEqual(df.loc[1, "count2"], 17)
 
-
     def test_where_in(self):
         pa_table = csv.read_csv(self.csv_file)
         test_table = self.session.import_table(pa_table)