11from io import BytesIO
2+ import ast
23import pickle
34
45import numpy as np
56import pytest
67
78import matplotlib as mpl
89from matplotlib import cm
10+ from matplotlib .testing import subprocess_run_helper
911from matplotlib .testing .decorators import check_figures_equal
1012from matplotlib .dates import rrulewrapper
1113from matplotlib .lines import VertexSelector
@@ -42,9 +44,7 @@ def test_simple():
4244pickle .dump (fig ,BytesIO (),pickle .HIGHEST_PROTOCOL )
4345
4446
45- @mpl .style .context ("default" )
46- @check_figures_equal (extensions = ["png" ])
47- def test_complete (fig_test ,fig_ref ):
47+ def _generate_complete_test_figure (fig_ref ):
4848fig_ref .set_size_inches ((10 ,6 ))
4949plt .figure (fig_ref )
5050
@@ -83,12 +83,17 @@ def test_complete(fig_test, fig_ref):
8383plt .quiver (x ,y ,u ,v )
8484
8585plt .subplot (3 ,3 ,8 )
86- plt .scatter (x ,x ** 2 ,label = '$x^2$' )
86+ plt .scatter (x ,x ** 2 ,label = '$x^2$' )
8787plt .legend (loc = 'upper left' )
8888
8989plt .subplot (3 ,3 ,9 )
9090plt .errorbar (x ,x * - 0.5 ,xerr = 0.2 ,yerr = 0.4 )
9191
92+
93+ @mpl .style .context ("default" )
94+ @check_figures_equal (extensions = ["png" ])
95+ def test_complete (fig_test ,fig_ref ):
96+ _generate_complete_test_figure (fig_ref )
9297# plotting is done, now test its pickle-ability
9398pkl = BytesIO ()
9499pickle .dump (fig_ref ,pkl ,pickle .HIGHEST_PROTOCOL )
@@ -101,6 +106,46 @@ def test_complete(fig_test, fig_ref):
101106plt .close (loaded )
102107
103108
109+ def _pickle_load_subprocess ():
110+ import os
111+ import pickle
112+
113+ path = os .environ ['PICKLE_FILE_PATH' ]
114+
115+ with open (path ,'rb' )as blob :
116+ fig = pickle .load (blob )
117+
118+ print (str (pickle .dumps (fig )))
119+
120+
121+ @mpl .style .context ("default" )
122+ @check_figures_equal (extensions = ['png' ])
123+ def test_pickle_load_from_subprocess (fig_test ,fig_ref ,tmp_path ):
124+ _generate_complete_test_figure (fig_ref )
125+
126+ fp = tmp_path / 'sinus.pickle'
127+ assert not fp .exists ()
128+
129+ with fp .open ('wb' )as file :
130+ pickle .dump (fig_ref ,file ,pickle .HIGHEST_PROTOCOL )
131+ assert fp .exists ()
132+
133+ proc = subprocess_run_helper (
134+ _pickle_load_subprocess ,
135+ timeout = 60 ,
136+ extra_env = {'PICKLE_FILE_PATH' :str (fp )}
137+ )
138+
139+ loaded_fig = pickle .loads (ast .literal_eval (proc .stdout ))
140+
141+ loaded_fig .canvas .draw ()
142+
143+ fig_test .set_size_inches (loaded_fig .get_size_inches ())
144+ fig_test .figimage (loaded_fig .canvas .renderer .buffer_rgba ())
145+
146+ plt .close (loaded_fig )
147+
148+
104149def test_gcf ():
105150fig = plt .figure ("a label" )
106151buf = BytesIO ()