Enable floating parameters in SET (ex and px) (#1635)

Author: AvitalFineRedis

redis/commands.py

@@ -1174,12 +1174,18 @@ def set(self, name, value,
             pieces.append('EX')
             if isinstance(ex, datetime.timedelta):
                 ex = int(ex.total_seconds())
-            pieces.append(ex)
+            if isinstance(ex, int):
+                pieces.append(ex)
+            else:
+                raise DataError("ex must be datetime.timedelta or int")
         if px is not None:
             pieces.append('PX')
             if isinstance(px, datetime.timedelta):
                 px = int(px.total_seconds() * 1000)
-            pieces.append(px)
+            if isinstance(px, int):
+                pieces.append(px)
+            else:
+                raise DataError("px must be datetime.timedelta or int")
         if exat is not None:
             pieces.append('EXAT')
             if isinstance(exat, datetime.datetime):

tests/test_commands.py

@@ -1104,6 +1104,8 @@ def test_set_px(self, r):
         assert r['a'] == b'1'
         assert 0 < r.pttl('a') <= 10000
         assert 0 < r.ttl('a') <= 10
+        with pytest.raises(exceptions.DataError):
+            assert r.set('a', '1', px=10.0)
 
     @skip_if_server_version_lt('2.6.0')
     def test_set_px_timedelta(self, r):
@@ -1116,6 +1118,8 @@ def test_set_px_timedelta(self, r):
     def test_set_ex(self, r):
         assert r.set('a', '1', ex=10)
         assert 0 < r.ttl('a') <= 10
+        with pytest.raises(exceptions.DataError):
+            assert r.set('a', '1', ex=10.0)
 
     @skip_if_server_version_lt('2.6.0')
     def test_set_ex_timedelta(self, r):