Skip to content

Commit 68fd2ee

Browse files
authored
Fix doctests by ignoring warning from netcdf (#10599)
1 parent d237a71 commit 68fd2ee

File tree

2 files changed

+13
-5
lines changed

2 files changed

+13
-5
lines changed

xarray/tests/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@
4848
warnings.filterwarnings("ignore", "'urllib3.contrib.pyopenssl' module is deprecated")
4949
warnings.filterwarnings("ignore", "Deprecated call to `pkg_resources.declare_namespace")
5050
warnings.filterwarnings("ignore", "pkg_resources is deprecated as an API")
51+
warnings.filterwarnings("ignore", message="numpy.ndarray size changed")
5152

5253
arm_xfail = pytest.mark.xfail(
5354
platform.machine() == "aarch64" or "arm" in platform.machine(),

xarray/tests/test_plot.py

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -331,9 +331,10 @@ def test_2d_line_accepts_legend_kw(self) -> None:
331331
assert not plt.gca().get_legend()
332332
plt.cla()
333333
self.darray[:, :, 0].plot.line(x="dim_0", add_legend=True)
334-
assert plt.gca().get_legend()
334+
legend = plt.gca().get_legend()
335+
assert legend is not None
335336
# check whether legend title is set
336-
assert plt.gca().get_legend().get_title().get_text() == "dim_1"
337+
assert legend.get_title().get_text() == "dim_1"
337338

338339
def test_2d_line_accepts_x_kw(self) -> None:
339340
self.darray[:, :, 0].plot.line(x="dim_0")
@@ -344,10 +345,14 @@ def test_2d_line_accepts_x_kw(self) -> None:
344345

345346
def test_2d_line_accepts_hue_kw(self) -> None:
346347
self.darray[:, :, 0].plot.line(hue="dim_0")
347-
assert plt.gca().get_legend().get_title().get_text() == "dim_0"
348+
legend = plt.gca().get_legend()
349+
assert legend is not None
350+
assert legend.get_title().get_text() == "dim_0"
348351
plt.cla()
349352
self.darray[:, :, 0].plot.line(hue="dim_1")
350-
assert plt.gca().get_legend().get_title().get_text() == "dim_1"
353+
legend = plt.gca().get_legend()
354+
assert legend is not None
355+
assert legend.get_title().get_text() == "dim_1"
351356

352357
def test_2d_coords_line_plot(self) -> None:
353358
lon, lat = np.meshgrid(np.linspace(-20, 20, 5), np.linspace(0, 30, 4))
@@ -2917,7 +2922,9 @@ def test_legend_labels(self) -> None:
29172922
pc = ds2.plot.scatter(x="A", y="B", markersize="hue")
29182923
axes = pc.axes
29192924
assert axes is not None
2920-
actual = [t.get_text() for t in axes.get_legend().texts]
2925+
legend = axes.get_legend()
2926+
assert legend is not None
2927+
actual = [t.get_text() for t in legend.texts]
29212928
expected = ["hue", "a", "b"]
29222929
assert actual == expected
29232930

0 commit comments

Comments
 (0)