added tests for addhash and aggregations

Author: dipinarora9

redisearch/reducers.py

@@ -56,6 +56,7 @@ class avg(FieldOnlyReducer):
     def __init__(self, field):
         super(avg, self).__init__(field)
 
+
 class tolist(FieldOnlyReducer):
     """
     Returns all the matched properties in a list
@@ -65,6 +66,7 @@ class tolist(FieldOnlyReducer):
     def __init__(self, field):
         super(tolist, self).__init__(field)
 
+
 class count_distinct(FieldOnlyReducer):
     """
     Calculate the number of distinct values contained in all the results in
@@ -82,7 +84,7 @@ class count_distinctish(FieldOnlyReducer):
     group for the given field. This uses a faster algorithm than
     `count_distinct` but is less accurate
     """
-    name = 'COUNT_DISTINCTISH'
+    NAME = 'COUNT_DISTINCTISH'
 
 
 class quantile(Reducer):
@@ -101,7 +103,7 @@ class stddev(FieldOnlyReducer):
     """
     Return the standard deviation for the values within the group
     """
-    name = 'STDDEV'
+    NAME = 'STDDEV'
 
     def __init__(self, field):
         super(stddev, self).__init__(field)

test/test.py

@@ -11,6 +11,8 @@
 import six
 
 from redisearch import *
+import redisearch.aggregation as aggregations
+import redisearch.reducers as reducers
 
 WILL_PLAY_TEXT = os.path.abspath(os.path.dirname(__file__)) + '/will_play_text.csv.bz2'
 
@@ -171,6 +173,33 @@ def getCleanClient(self, name):
             pass
 
         return client
+    
+    def testAddHash(self):
+        conn = self.redis()
+
+        # Creating a client with a given index name
+        client = Client('idx', port=conn.port)
+
+        client.redis.flushdb()
+        # Creating the index definition and schema
+        client.create_index((TextField('title',
+                                       weight=5.0), TextField('body')))
+        redis_client = redis.Redis(port=conn.port)
+        redis_client.hset(
+            'doc1',
+            mapping={
+                'title': 'RediSearch',
+                'body': 'Redisearch impements a search engine on top of redis'
+            })
+        # Indexing the hash
+        client.add_document_hash('doc1')
+
+        # Searching with complext parameters:
+        q = Query("search engine").verbatim().no_content().paging(0, 5)
+
+        res = client.search(q)
+
+        self.assertTrue('doc1', res.docs[0].id)
 
     def testPayloads(self):
 
@@ -658,6 +687,72 @@ def testGet(self):
         self.assertEqual([['f1', 'some valid content dd2', 'f2', 'this is sample text ff2']], client.get('doc2'))
         self.assertEqual([['f1', 'some valid content dd1', 'f2', 'this is sample text ff1'], ['f1', 'some valid content dd2', 'f2', 'this is sample text ff2']], client.get('doc1', 'doc2'))
 
+    def testAggregations(self):
+        conn = self.redis()
+        client = Client('myIndex', port=conn.port)
+        client.redis.flushdb()
+
+        # Creating the index definition and schema
+        client.create_index((NumericField('random_num'), TextField('title'),
+                             TextField('body'), TextField('parent')))
+
+        # Indexing a document
+        client.add_document(
+            'search',
+            title='RediSearch',
+            body='Redisearch impements a search engine on top of redis',
+            parent='redis',
+            random_num=10)
+        client.add_document(
+            'ai',
+            title='RedisAI',
+            body=
+            'RedisAI executes Deep Learning/Machine Learning models and managing their data.',
+            parent='redis',
+            random_num=3)
+        client.add_document(
+            'json',
+            title='RedisJson',
+            body=
+            'RedisJSON implements ECMA-404 The JSON Data Interchange Standard as a native data type.',
+            parent='redis',
+            random_num=8)
+
+        req = aggregations.AggregateRequest('redis').group_by(
+            "@parent",
+            reducers.count(),
+            reducers.count_distinct('@title'),
+            reducers.count_distinctish('@title'),
+            reducers.sum("@random_num"),
+            reducers.min("@random_num"),
+            reducers.max("@random_num"),
+            reducers.avg("@random_num"),
+            reducers.stddev("random_num"),
+            reducers.quantile("@random_num", 0.5),
+            reducers.tolist("@title"),
+            reducers.first_value("@title"),
+            reducers.random_sample("@title", 2),
+        )
+
+        res = client.aggregate(req)
+
+        res = res.rows[0]
+
+        self.assertEqual(len(res), 26)
+        self.assertEqual(b'redis', res[1])
+        self.assertEqual(b'3', res[3])
+        self.assertEqual(b'3', res[5])
+        self.assertEqual(b'3', res[7])
+        self.assertEqual(b'21', res[9])
+        self.assertEqual(b'3', res[11])
+        self.assertEqual(b'10', res[13])
+        self.assertEqual(b'7', res[15])
+        self.assertEqual(b'3.60555127546', res[17])
+        self.assertEqual(b'10', res[19])
+        self.assertEqual([b'RediSearch', b'RedisAI', b'RedisJson'], res[21])
+        self.assertEqual(b'RediSearch', res[23])
+        self.assertEqual(2, len(res[25]))
+
 
 if __name__ == '__main__':