diff --git a/corrai/sensitivity.py b/corrai/sensitivity.py index 365ef23..c415762 100644 --- a/corrai/sensitivity.py +++ b/corrai/sensitivity.py @@ -425,7 +425,7 @@ def salib_plot_bar( data=res[sensitivity_metric], index=[par.name for par in self.sampler.sample.parameters], name=f"{sensitivity_metric} {unit}", - ), + ).sort_values(), title=title, ) diff --git a/tests/test_sensitivity.py b/tests/test_sensitivity.py index 9141d79..e50bd4e 100644 --- a/tests/test_sensitivity.py +++ b/tests/test_sensitivity.py @@ -226,12 +226,11 @@ def test_morris_plots(self): fig_bar["layout"]["title"]["text"] == "Morris euclidian_distance mean res" ) - res_euclidian = morris_analysis.analyze("res")["mean_res"] fig_data = fig_bar.data[0] - expected_x = [p.name for p in PARAMETER_LIST] - expected_y = res_euclidian["euclidian_distance"].tolist() - assert list(fig_data.x) == expected_x - np.testing.assert_allclose(fig_data.y, expected_y) + assert list(fig_data.x) == ["par_x1", "par_x3", "par_x2"] + np.testing.assert_allclose( + fig_data.y, [1.455258008259737, 10.823232337117245, 13.63990010960599] + ) fig_dyn = morris_analysis.plot_dynamic_metric( indicator="res",