fix:test

cunla · May 6, 2024 · 9e8d421 · 9e8d421
1 parent d602ecf
commit 9e8d421
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 156 deletions.
diff --git a/fakeredis/_fakesocket.py b/fakeredis/_fakesocket.py
@@ -1,6 +1,9 @@
 from typing import Optional, Set
 
-from fakeredis.stack import JSONCommandsMixin, BFCommandsMixin, CFCommandsMixin, CMSCommandsMixin, TopkCommandsMixin
+from fakeredis.stack import (
+    JSONCommandsMixin, BFCommandsMixin, CFCommandsMixin, CMSCommandsMixin, TopkCommandsMixin,
+    TDigestCommandsMixin,
+)
 from ._basefakesocket import BaseFakeSocket
 from .commands_mixins.bitmap_mixin import BitmapCommandsMixin
 from .commands_mixins.connection_mixin import ConnectionCommandsMixin
@@ -9,7 +12,6 @@
 from .commands_mixins.hash_mixin import HashCommandsMixin
 from .commands_mixins.list_mixin import ListCommandsMixin
 from .commands_mixins.pubsub_mixin import PubSubCommandsMixin
-from .stack._tdigest_mixin import TDigestCommandsMixin
 
 try:
     from .commands_mixins.scripting_mixin import ScriptingCommandsMixin

diff --git a/fakeredis/stack/__init__.py b/fakeredis/stack/__init__.py
@@ -1,3 +1,4 @@
+from ._tdigest_mixin import TDigestCommandsMixin
 from ._topk_mixin import TopkCommandsMixin  # noqa: F401
 
 try:
@@ -40,4 +41,6 @@ class CMSCommandsMixin:  # type: ignore # noqa: E303
     "JSONObject",
     "BFCommandsMixin",
     "CFCommandsMixin",
-    "CMSCommandsMixin", ]
+    "CMSCommandsMixin",
+    "TDigestCommandsMixin",
+]
diff --git a/fakeredis/stack/_tdigest.py b/fakeredis/stack/_tdigest.py