SMISMEMBER support (#1667)

Author: chayim

redis/commands/core.py

@@ -1768,6 +1768,14 @@ def smembers(self, name):
         """Return all members of the set ``name``"""
         return self.execute_command('SMEMBERS', name)
 
+    def smismember(self, name, values, *args):
+        """
+        Return whether each value in ``values`` is a member of the set ``name``
+        as a list of ``bool`` in the order of ``values``
+        """
+        args = list_or_args(values, args)
+        return self.execute_command('SMISMEMBER', name, *args)
+
     def smove(self, src, dst, value):
         """Move ``value`` from set ``src`` to set ``dst`` atomically"""
         return self.execute_command('SMOVE', src, dst, value)

tests/test_commands.py

@@ -1577,6 +1577,13 @@ def test_smembers(self, r):
         r.sadd('a', '1', '2', '3')
         assert r.smembers('a') == {b'1', b'2', b'3'}
 
+    @skip_if_server_version_lt('6.2.0')
+    def test_smismember(self, r):
+        r.sadd('a', '1', '2', '3')
+        result_list = [True, False, True, True]
+        assert r.smismember('a', '1', '4', '2', '3') == result_list
+        assert r.smismember('a', ['1', '4', '2', '3']) == result_list
+
     def test_smove(self, r):
         r.sadd('a', 'a1', 'a2')
         r.sadd('b', 'b1', 'b2')