Fix GRAPH.LIST & TDIGEST.QUANTILE tests (#2335)

* fix graph list_keys test

* async test

* fix tdigest.quantile

* fix

Author: dvora-h

tests/test_asyncio/test_bloom.py

@@ -366,21 +366,21 @@ async def test_tdigest_quantile(modclient: redis.Redis):
     # assert min min/max have same result as quantile 0 and 1
     assert (
         await modclient.tdigest().max("tDigest")
-        == (await modclient.tdigest().quantile("tDigest", 1.0))[1]
+        == (await modclient.tdigest().quantile("tDigest", 1.0))[0]
     )
     assert (
         await modclient.tdigest().min("tDigest")
-        == (await modclient.tdigest().quantile("tDigest", 0.0))[1]
+        == (await modclient.tdigest().quantile("tDigest", 0.0))[0]
     )
 
-    assert 1.0 == round((await modclient.tdigest().quantile("tDigest", 0.01))[1], 2)
-    assert 99.0 == round((await modclient.tdigest().quantile("tDigest", 0.99))[1], 2)
+    assert 1.0 == round((await modclient.tdigest().quantile("tDigest", 0.01))[0], 2)
+    assert 99.0 == round((await modclient.tdigest().quantile("tDigest", 0.99))[0], 2)
 
     # test multiple quantiles
     assert await modclient.tdigest().create("t-digest", 100)
     assert await modclient.tdigest().add("t-digest", [1, 2, 3, 4, 5], [1.0] * 5)
     res = await modclient.tdigest().quantile("t-digest", 0.5, 0.8)
-    assert [0.5, 3.0, 0.8, 5.0] == res
+    assert [3.0, 5.0] == res
 
 
 @pytest.mark.redismod

tests/test_asyncio/test_graph.py

@@ -348,11 +348,11 @@ async def test_list_keys(modclient: redis.Redis):
     result = await modclient.graph().list_keys()
     assert result == []
 
-    await modclient.execute_command("GRAPH.EXPLAIN", "G", "RETURN 1")
+    await modclient.graph("G").query("CREATE (n)")
     result = await modclient.graph().list_keys()
     assert result == ["G"]
 
-    await modclient.execute_command("GRAPH.EXPLAIN", "X", "RETURN 1")
+    await modclient.graph("X").query("CREATE (m)")
     result = await modclient.graph().list_keys()
     assert result == ["G", "X"]
 

tests/test_bloom.py

@@ -365,17 +365,17 @@ def test_tdigest_quantile(client):
     )
     # assert min min/max have same result as quantile 0 and 1
     res = client.tdigest().quantile("tDigest", 1.0)
-    assert client.tdigest().max("tDigest") == res[1]
+    assert client.tdigest().max("tDigest") == res[0]
     res = client.tdigest().quantile("tDigest", 0.0)
-    assert client.tdigest().min("tDigest") == res[1]
+    assert client.tdigest().min("tDigest") == res[0]
 
-    assert 1.0 == round(client.tdigest().quantile("tDigest", 0.01)[1], 2)
-    assert 99.0 == round(client.tdigest().quantile("tDigest", 0.99)[1], 2)
+    assert 1.0 == round(client.tdigest().quantile("tDigest", 0.01)[0], 2)
+    assert 99.0 == round(client.tdigest().quantile("tDigest", 0.99)[0], 2)
 
     # test multiple quantiles
     assert client.tdigest().create("t-digest", 100)
     assert client.tdigest().add("t-digest", [1, 2, 3, 4, 5], [1.0] * 5)
-    assert [0.5, 3.0, 0.8, 5.0] == client.tdigest().quantile("t-digest", 0.5, 0.8)
+    assert [3.0, 5.0] == client.tdigest().quantile("t-digest", 0.5, 0.8)
 
 
 @pytest.mark.redismod

tests/test_graph.py

@@ -366,11 +366,11 @@ def test_list_keys(client):
     result = client.graph().list_keys()
     assert result == []
 
-    client.execute_command("GRAPH.EXPLAIN", "G", "RETURN 1")
+    client.graph("G").query("CREATE (n)")
     result = client.graph().list_keys()
     assert result == ["G"]
 
-    client.execute_command("GRAPH.EXPLAIN", "X", "RETURN 1")
+    client.graph("X").query("CREATE (m)")
     result = client.graph().list_keys()
     assert result == ["G", "X"]