Zunion (#1522)

* zinter

* change options in _zaggregate

* skip for previous versions

* add client function

* validate the aggregate value

* change options to get

* add more aggregate tests

* add weights guidance

1 files changed, 20 insertions, 0 deletions

diff --git a/tests/test_commands.py b/tests/test_commands.py
index 9a618aa..3c87dd9 100644
--- a/tests/test_commands.py
+++ b/tests/test_commands.py
@@ -1813,6 +1813,26 @@ class TestRedisCommands:
         assert r.zscore('a', 'a2') == 2.0
         assert r.zscore('a', 'a4') is None
 
+    @skip_if_server_version_lt('6.2.0')
+    def test_zunion(self, r):
+        r.zadd('a', {'a1': 1, 'a2': 1, 'a3': 1})
+        r.zadd('b', {'a1': 2, 'a2': 2, 'a3': 2})
+        r.zadd('c', {'a1': 6, 'a3': 5, 'a4': 4})
+        # sum
+        assert r.zunion(['a', 'b', 'c']) == \
+            [b'a2', b'a4', b'a3', b'a1']
+        assert r.zunion(['a', 'b', 'c'], withscores=True) == \
+            [(b'a2', 3), (b'a4', 4), (b'a3', 8), (b'a1', 9)]
+        # max
+        assert r.zunion(['a', 'b', 'c'], aggregate='MAX', withscores=True)\
+               == [(b'a2', 2), (b'a4', 4), (b'a3', 5), (b'a1', 6)]
+        # min
+        assert r.zunion(['a', 'b', 'c'], aggregate='MIN', withscores=True)\
+               == [(b'a1', 1), (b'a2', 1), (b'a3', 1), (b'a4', 4)]
+        # with weight
+        assert r.zunion({'a': 1, 'b': 2, 'c': 3}, withscores=True)\
+               == [(b'a2', 5), (b'a4', 12), (b'a3', 20), (b'a1', 23)]
+
     def test_zunionstore_sum(self, r):
         r.zadd('a', {'a1': 1, 'a2': 1, 'a3': 1})
         r.zadd('b', {'a1': 2, 'a2': 2, 'a3': 2})