Add suggestions to MovieWriterRegistry lookup errors

QuLogic · QuLogic · commit 71420d28f1e6 · 2026-04-17T13:59:46.000-04:00
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -117,6 +117,7 @@ def list(self):
 
     def __getitem__(self, name):
         """Get an available writer class from its name."""
+        _api.check_in_list(self._registered, writer=name)
         if self.is_available(name):
             return self._registered[name]
         raise RuntimeError(f"Requested MovieWriter ({name}) not available")
diff --git a/lib/matplotlib/tests/test_animation.py b/lib/matplotlib/tests/test_animation.py
@@ -44,6 +44,14 @@ def animate(i):
     return klass(fig=fig, func=animate, init_func=init, **kwargs)
 
 
+def test_invalid_writer():
+    # Note, this triggers for Animation.save as well, but this is a lighter test.
+    with pytest.raises(ValueError,
+                       match=r"'pllow' is not a valid value for writer\. "
+                             r"Did you mean: 'pillow'\?"):
+        animation.writers['pllow']
+
+
 class NullMovieWriter(animation.AbstractMovieWriter):
     """
     A minimal MovieWriter.  It doesn't actually write anything.
