@@ -331,9 +331,10 @@ def test_2d_line_accepts_legend_kw(self) -> None:
331
331
assert not plt .gca ().get_legend ()
332
332
plt .cla ()
333
333
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
335
336
# 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"
337
338
338
339
def test_2d_line_accepts_x_kw (self ) -> None :
339
340
self .darray [:, :, 0 ].plot .line (x = "dim_0" )
@@ -344,10 +345,14 @@ def test_2d_line_accepts_x_kw(self) -> None:
344
345
345
346
def test_2d_line_accepts_hue_kw (self ) -> None :
346
347
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"
348
351
plt .cla ()
349
352
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"
351
356
352
357
def test_2d_coords_line_plot (self ) -> None :
353
358
lon , lat = np .meshgrid (np .linspace (- 20 , 20 , 5 ), np .linspace (0 , 30 , 4 ))
@@ -2917,7 +2922,9 @@ def test_legend_labels(self) -> None:
2917
2922
pc = ds2 .plot .scatter (x = "A" , y = "B" , markersize = "hue" )
2918
2923
axes = pc .axes
2919
2924
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 ]
2921
2928
expected = ["hue" , "a" , "b" ]
2922
2929
assert actual == expected
2923
2930
0 commit comments