Added tests for filtered aggregation

druid-io · Sep 18, 2015 · 8b8b138 · 8b8b138
1 parent 85bc393
commit 8b8b138
Showing 1 changed file with 69 additions and 0 deletions.
diff --git a/tests/utils/test_aggregators.py b/tests/utils/test_aggregators.py
@@ -1,8 +1,10 @@
 # -*- coding: UTF-8 -*-
 
 from operator import itemgetter
+from copy import deepcopy
 
 from pydruid.utils import aggregators
+from pydruid.utils import filters
 
 
 class TestAggregators:
@@ -16,6 +18,27 @@ def test_aggregators(self):
         for f, agg_type in aggs_funcs:
             assert f('metric') == {'type': agg_type, 'fieldName': 'metric'}
 
+    def test_filtered_aggregator(self):
+        filter_ = filters.Filter(dimension='dim', value='val')
+        aggs = [aggregators.count('metric1'),
+                aggregators.longsum('metric2'),
+                aggregators.doublesum('metric3'),
+                aggregators.min('metric4'),
+                aggregators.max('metric5'),
+                aggregators.hyperunique('metric6')]
+        for agg in aggs:
+            expected = {
+                'type': 'filtered',
+                'filter': {
+                    'type': 'selector',
+                    'dimension': 'dim',
+                    'value': 'val'
+                },
+                'aggregator': agg
+            }
+            actual = aggregators.filtered(filter_, agg)
+            assert actual == expected
+
     def test_build_aggregators(self):
         agg_input = {
             'agg1': aggregators.count('metric1'),
@@ -36,3 +59,49 @@ def test_build_aggregators(self):
         ]
         assert (sorted(built_agg, key=itemgetter('name')) ==
                 sorted(expected, key=itemgetter('name')))
+
+    def test_build_filtered_aggregator(self):
+        filter_ = filters.Filter(dimension='dim', value='val')
+        agg_input = {
+            'agg1': aggregators.filtered(filter_,
+                                         aggregators.count('metric1')),
+            'agg2': aggregators.filtered(filter_,
+                                         aggregators.longsum('metric2')),
+            'agg3': aggregators.filtered(filter_,
+                                         aggregators.doublesum('metric3')),
+            'agg4': aggregators.filtered(filter_,
+                                         aggregators.min('metric4')),
+            'agg5': aggregators.filtered(filter_,
+                                         aggregators.max('metric5')),
+            'agg6': aggregators.filtered(filter_,
+                                         aggregators.hyperunique('metric6'))
+        }
+        base = {
+            'type': 'filtered',
+            'filter': {
+                'type': 'selector',
+                'dimension': 'dim',
+                'value': 'val'
+            }
+        }
+
+        aggs = [
+            {'name': 'agg1', 'type': 'count', 'fieldName': 'metric1'},
+            {'name': 'agg2', 'type': 'longSum', 'fieldName': 'metric2'},
+            {'name': 'agg3', 'type': 'doubleSum', 'fieldName': 'metric3'},
+            {'name': 'agg4', 'type': 'min', 'fieldName': 'metric4'},
+            {'name': 'agg5', 'type': 'max', 'fieldName': 'metric5'},
+            {'name': 'agg6', 'type': 'hyperUnique', 'fieldName': 'metric6'},
+        ]
+        expected = []
+        for agg in aggs:
+            exp = deepcopy(base)
+            exp.update({'aggregator': agg})
+            expected.append(exp)
+
+        built_agg = aggregators.build_aggregators(agg_input)
+        expected = sorted(built_agg, key=lambda k: itemgetter('name')(
+            itemgetter('aggregator')(k)))
+        actual = sorted(expected, key=lambda k: itemgetter('name')(
+            itemgetter('aggregator')(k)))
+        assert expected == actual