Add count_where into the python rollup test.

deephaven · Dec 20, 2024 · a9ffa4f · a9ffa4f
1 parent d348e3f
commit a9ffa4f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/py/server/tests/test_rollup_tree_table.py b/py/server/tests/test_rollup_tree_table.py
@@ -4,7 +4,7 @@
 import unittest
 
 from deephaven import read_csv, empty_table
-from deephaven.agg import sum_, avg, count_, first, last, max_, min_, std, abs_sum, \
+from deephaven.agg import sum_, avg, count_, count_where, first, last, max_, min_, std, abs_sum, \
     var
 from deephaven.filters import Filter
 from deephaven.table import NodeType
@@ -18,6 +18,7 @@ def setUp(self):
         self.aggs_for_rollup = [
             avg(["aggAvg=var"]),
             count_("aggCount"),
+            count_where("aggCountWhere", "var > 0"),
             first(["aggFirst=var"]),
             last(["aggLast=var"]),
             max_(["aggMax=var"]),