Apr 22, 2024 · Apr 23, 2024
diff --git a/lib/matplotlib/cm.py b/lib/matplotlib/cm.py
 """

 from collections.abc import Mapping
 import difflib
 import functools

 import numpy as np
        try:
            return self._cmaps[item].copy()
        except KeyError:
            raise KeyError(f"{item!r} is not a known colormap name") from None
            known = list(self._cmaps)
            if len(best := difflib.get_close_matches(item, known, cutoff=.1)):
                msg = f"Did you mean one of {best}?"
            else:
                msg = ""
            raise KeyError(f"{item!r} is not a known colormap name. {msg}") from None

    def __iter__(self):
        return iter(self._cmaps)
diff --git a/lib/matplotlib/tests/test_colors.py b/lib/matplotlib/tests/test_colors.py
    assert_array_equal(
        to_rgba_array(c, alpha), [[0., 0., 1., 1.], [0., 0., 0., 0.]]
    )


 def test_close_error_name():
    with pytest.raises(KeyError) as exinfo:
        matplotlib.colormaps["grays"]

    msg = exinfo.value.args[0]

    assert "Grays" in msg
    assert "gray" in msg
    assert "gray_r" in msg
Original file line number	Diff line number	Diff line change
Expand Up		@@ -15,6 +15,7 @@
		"""

		from collections.abc import Mapping
		import difflib
		import functools

		import numpy as np
Expand DownExpand Up		@@ -88,7 +89,12 @@
		try:
		return self._cmaps[item].copy()
		except KeyError:
		raise KeyError(f"{item!r} is not a known colormap name") from None
		known = list(self._cmaps)
		if len(best := difflib.get_close_matches(item, known, cutoff=.1)):
		msg = f"Did you mean one of {best}?"
		else:
		msg = ""
Check warning on line 96 in lib/matplotlib/cm.py View check run for this annotation Codecov/ codecov/patch lib/matplotlib/cm.py#L96 `Added line #L96 was not covered by tests`
		raise KeyError(f"{item!r} is not a known colormap name. {msg}") from None

		def __iter__(self):
		return iter(self._cmaps)
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -1697,3 +1697,14 @@ def test_to_rgba_array_none_color_with_alpha_param():
		assert_array_equal(
		to_rgba_array(c, alpha), [[0., 0., 1., 1.], [0., 0., 0., 0.]]
		)


		def test_close_error_name():
		with pytest.raises(KeyError) as exinfo:
		matplotlib.colormaps["grays"]

		msg = exinfo.value.args[0]

		assert "Grays" in msg
		assert "gray" in msg
		assert "gray_r" in msg