From 4abe5ecceda8878324adff424f9375159d2dd4a6 Mon Sep 17 00:00:00 2001 From: Gregory Petrochenkov Date: Fri, 7 Jul 2023 00:56:11 -0400 Subject: [PATCH] Update plot tests --- tests/test_accessors.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) 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