diff --git a/tests/test_accessors.py b/tests/test_accessors.py index 05e0c13a..122f8663 100644 --- a/tests/test_accessors.py +++ b/tests/test_accessors.py @@ -241,7 +241,7 @@ def test_data_frame_accessor_compute_metrics( ) def test_categorical_plot_success(candidate_map, crs, entries): candidate_map.rio.set_crs(crs) - viz_object = candidate_map.gval.cat_plot() + viz_object = candidate_map.gval.cat_plot(basemap=None) assert len(viz_object.axes.get_legend().texts) == entries @@ -252,7 +252,7 @@ def test_categorical_plot_success(candidate_map, crs, entries): def test_categorical_plot_fail(candidate_map, legend_labels, num_classes): candidate_map.data = np.random.choice(np.arange(num_classes), candidate_map.shape) with raises(ValueError): - _ = candidate_map.gval.cat_plot(legend_labels=legend_labels) + _ = candidate_map.gval.cat_plot(legend_labels=legend_labels, basemap=None) @parametrize_with_cases( @@ -260,7 +260,8 @@ def test_categorical_plot_fail(candidate_map, legend_labels, num_classes): glob="continuous_plot_success", ) def test_continuous_plot_success(candidate_map, axes): - viz_object = candidate_map.gval.cont_plot() + args = {} if axes == 2 else {"basemap": None} + viz_object = candidate_map.gval.cont_plot(**args) assert len(viz_object.figure.axes) == axes