Fix __rich_repr__ type annotation

jcrist · Oct 3, 2023 · b7fa0c2 · b7fa0c2
1 parent 256d84a
commit b7fa0c2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/msgspec/__init__.pyi b/msgspec/__init__.pyi
@@ -68,7 +68,9 @@ class Struct:
         weakref: bool = False,
         dict: bool = False,
     ) -> None: ...
-    def __rich_repr__(self) -> Iterable[Tuple[str, Any]]: ...
+    def __rich_repr__(
+        self,
+    ) -> Iterable[Union[Any, Tuple[Any], Tuple[str, Any], Tuple[str, Any, Any]]]: ...
 
 def defstruct(
     name: str,

diff --git a/tests/basic_typing_examples.py b/tests/basic_typing_examples.py
@@ -311,7 +311,10 @@ class Point(msgspec.Struct):
     a.x = a.x + b.y
     repr(a)
 
-    for name, val in a.__rich_repr__():
+    for item in a.__rich_repr__():
+        assert isinstance(item, tuple)
+        assert len(item) == 2
+        name, val = item
         print(f"{name} = {val}")