Fix __rich_repr__ type annotation

jcrist · Oct 3, 2023 · 792d8ea · 792d8ea
1 parent d3abf75
commit 792d8ea
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}")