Support TDIGEST.MERGESTORE and make compression optional on TDIGEST.CREATE (#2319)

* support 2.4

* async test

* skip tests

* linters

1 files changed, 15 insertions, 1 deletions

diff --git a/redis/commands/bf/commands.py b/redis/commands/bf/commands.py
index 7d36b93..84a6b5f 100644
--- a/redis/commands/bf/commands.py
+++ b/redis/commands/bf/commands.py
@@ -49,6 +49,7 @@ TDIGEST_QUANTILE = "TDIGEST.QUANTILE"
 TDIGEST_MIN = "TDIGEST.MIN"
 TDIGEST_MAX = "TDIGEST.MAX"
 TDIGEST_INFO = "TDIGEST.INFO"
+TDIGEST_MERGESTORE = "TDIGEST.MERGESTORE"
 
 
 class BFCommands:
@@ -344,7 +345,7 @@ class TOPKCommands:
 
 
 class TDigestCommands:
-    def create(self, key, compression):
+    def create(self, key, compression=100):
         """
         Allocate the memory and initialize the t-digest.
         For more information see `TDIGEST.CREATE <https://redis.io/commands/tdigest.create>`_.
@@ -417,6 +418,19 @@ class TDigestCommands:
         """  # noqa
         return self.execute_command(TDIGEST_INFO, key)
 
+    def mergestore(self, dest_key, numkeys, *sourcekeys, compression=False):
+        """
+        Merges all of the values from `sourcekeys` keys to `dest_key` sketch.
+        If destination already exists, it is overwritten.
+
+
+        For more information see `TDIGEST.MERGESTORE <https://redis.io/commands/tdigest.mergestore>`_.
+        """  # noqa
+        params = [dest_key, numkeys, *sourcekeys]
+        if compression:
+            params.extend(["COMPRESSION", compression])
+        return self.execute_command(TDIGEST_MERGESTORE, *params)
+
 
 class CMSCommands:
     """Count-Min Sketch Commands"""