| import os |
|
|
| import pytest |
| from unittest import mock |
|
|
| import matplotlib as mpl |
| import matplotlib.pyplot as plt |
| try: |
| from matplotlib.backends import _macosx |
| except ImportError: |
| pytest.skip("These are mac only tests", allow_module_level=True) |
|
|
|
|
| @pytest.mark.backend('macosx') |
| def test_cached_renderer(): |
| |
| |
| fig = plt.figure(1) |
| fig.canvas.draw() |
| assert fig.canvas.get_renderer()._renderer is not None |
|
|
| fig = plt.figure(2) |
| fig.draw_without_rendering() |
| assert fig.canvas.get_renderer()._renderer is not None |
|
|
|
|
| @pytest.mark.backend('macosx') |
| def test_savefig_rcparam(monkeypatch, tmp_path): |
|
|
| def new_choose_save_file(title, directory, filename): |
| |
| |
| assert directory == str(tmp_path) |
| os.makedirs(f"{directory}/test") |
| return f"{directory}/test/{filename}" |
|
|
| monkeypatch.setattr(_macosx, "choose_save_file", new_choose_save_file) |
| fig = plt.figure() |
| with mpl.rc_context({"savefig.directory": tmp_path}): |
| fig.canvas.toolbar.save_figure() |
| |
| save_file = f"{tmp_path}/test/{fig.canvas.get_default_filename()}" |
| assert os.path.exists(save_file) |
|
|
| |
| |
| assert mpl.rcParams["savefig.directory"] == f"{tmp_path}/test" |
|
|
|
|
| @pytest.mark.backend('macosx') |
| def test_ipython(): |
| from matplotlib.testing import ipython_in_subprocess |
| ipython_in_subprocess("osx", {(8, 24): "macosx", (7, 0): "MacOSX"}) |
|
|
|
|
| @pytest.mark.backend('macosx') |
| def test_save_figure_return(): |
| fig, ax = plt.subplots() |
| ax.imshow([[1]]) |
| prop = "matplotlib.backends._macosx.choose_save_file" |
| with mock.patch(prop, return_value="foobar.png"): |
| fname = fig.canvas.manager.toolbar.save_figure() |
| os.remove("foobar.png") |
| assert fname == "foobar.png" |
| with mock.patch(prop, return_value=None): |
| fname = fig.canvas.manager.toolbar.save_figure() |
| assert fname is None |
|
|
