Add rounding to delicate timeseries test

Author: kristjanvalur

tests/test_asyncio/test_timeseries.py

@@ -217,6 +217,14 @@ async def test_range(modclient: redis.Redis):
     assert 10 == len(await modclient.ts().range(1, 0, 500, count=10))
 
 
+def round_reply(tuples, digits=3):
+    """
+    Round the timeseries reply since we are not testing for minute differences
+    in server implementation precision
+    """
+    return [(a, round(b, digits)) for (a, b) in tuples]
+
+
 @pytest.mark.redismod
 @skip_ifmodversion_lt("99.99.99", "timeseries")
 async def test_range_advanced(modclient: redis.Redis):
@@ -240,8 +248,10 @@ async def test_range_advanced(modclient: redis.Redis):
     assert [(0, 5.0), (5, 6.0)] == await modclient.ts().range(
         1, 0, 10, aggregation_type="count", bucket_size_msec=10, align=5
     )
-    assert [(0, 2.5500000000000003), (10, 3.0)] == await modclient.ts().range(
-        1, 0, 10, aggregation_type="twa", bucket_size_msec=10
+    assert [(0, 2.55), (10, 3.0)] == round_reply(
+        await modclient.ts().range(
+            1, 0, 10, aggregation_type="twa", bucket_size_msec=10
+        )
     )
 
 

tests/test_timeseries.py

@@ -210,6 +210,14 @@ def test_range(client):
     assert 10 == len(client.ts().range(1, 0, 500, count=10))
 
 
+def round_reply(tuples, digits=3):
+    """
+    Round the timeseries reply since we are not testing for minute differences
+    in server implementation precision
+    """
+    return [(a, round(b, digits)) for (a, b) in tuples]
+
+
 @pytest.mark.redismod
 @skip_ifmodversion_lt("99.99.99", "timeseries")
 def test_range_advanced(client):
@@ -233,8 +241,8 @@ def test_range_advanced(client):
     assert [(0, 5.0), (5, 6.0)] == client.ts().range(
         1, 0, 10, aggregation_type="count", bucket_size_msec=10, align=5
     )
-    assert [(0, 2.5500000000000003), (10, 3.0)] == client.ts().range(
-        1, 0, 10, aggregation_type="twa", bucket_size_msec=10
+    assert [(0, 2.55), (10, 3.0)] == round_reply(
+        client.ts().range(1, 0, 10, aggregation_type="twa", bucket_size_msec=10)
     )
 
 
@@ -343,8 +351,8 @@ def test_rev_range(client):
     assert [(1, 10.0), (0, 1.0)] == client.ts().revrange(
         1, 0, 10, aggregation_type="count", bucket_size_msec=10, align=1
     )
-    assert [(10, 3.0), (0, 2.5500000000000003)] == client.ts().revrange(
-        1, 0, 10, aggregation_type="twa", bucket_size_msec=10
+    assert [(10, 3.0), (0, 2.55)] == round_reply(
+        client.ts().revrange(1, 0, 10, aggregation_type="twa", bucket_size_msec=10)
     )