47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
|
import os
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
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():
|
||
|
|
# Make sure that figures have an associated renderer after
|
||
|
|
# a fig.canvas.draw() call
|
||
|
|
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):
|
||
|
|
# Replacement function instead of opening a GUI window
|
||
|
|
# Make a new directory for testing the update of the rcParams
|
||
|
|
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()
|
||
|
|
# Check the saved location got created
|
||
|
|
save_file = f"{tmp_path}/test/{fig.canvas.get_default_filename()}"
|
||
|
|
assert os.path.exists(save_file)
|
||
|
|
|
||
|
|
# Check the savefig.directory rcParam got updated because
|
||
|
|
# we added a subdirectory "test"
|
||
|
|
assert mpl.rcParams["savefig.directory"] == f"{tmp_path}/test"
|