Skip to content

Commit b466fc1

Browse files
committed
COMPAT: For MPL deprecation
1 parent c10ae89 commit b466fc1

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

pandas/tests/test_graphics.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -450,8 +450,9 @@ def _check_box_return_type(self, returned, return_type, expected_keys=None,
450450
self.assertIsInstance(value.lines, dict)
451451
elif return_type == 'dict':
452452
line = value['medians'][0]
453+
axes = line.axes if self.mpl_ge_1_5_0 else line.get_axes()
453454
if check_ax_title:
454-
self.assertEqual(line.get_axes().get_title(), key)
455+
self.assertEqual(axes.get_title(), key)
455456
else:
456457
raise AssertionError
457458

@@ -910,7 +911,7 @@ def test_hist_no_overlap(self):
910911
subplot(122)
911912
y.hist()
912913
fig = gcf()
913-
axes = fig.get_axes()
914+
axes = fig.axes if self.mpl_ge_1_5_0 else fig.get_axes()
914915
self.assertEqual(len(axes), 2)
915916

916917
@slow

pandas/tests/test_graphics_others.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ def test_hist_no_overlap(self):
150150
subplot(122)
151151
y.hist()
152152
fig = gcf()
153-
axes = fig.get_axes()
153+
axes = fig.axes if self.mpl_ge_1_5_0 else fig.get_axes()
154154
self.assertEqual(len(axes), 2)
155155

156156
@slow
@@ -242,11 +242,13 @@ def test_boxplot_legacy(self):
242242
# passed ax should be used:
243243
fig, ax = self.plt.subplots()
244244
axes = df.boxplot('Col1', by='X', ax=ax)
245-
self.assertIs(ax.get_axes(), axes)
245+
ax_axes = ax.axes if self.mpl_ge_1_5_0 else ax.get_axes()
246+
self.assertIs(ax_axes, axes)
246247

247248
fig, ax = self.plt.subplots()
248249
axes = df.groupby('Y').boxplot(ax=ax, return_type='axes')
249-
self.assertIs(ax.get_axes(), axes['A'])
250+
ax_axes = ax.axes if self.mpl_ge_1_5_0 else ax.get_axes()
251+
self.assertIs(ax_axes, axes['A'])
250252

251253
# Multiple columns with an ax argument should use same figure
252254
fig, ax = self.plt.subplots()

0 commit comments

Comments
 (0)