From 5f57429972d84770de96c052e34aa44959972bd8 Mon Sep 17 00:00:00 2001 From: Noe Brucy Date: Fri, 26 Mar 2021 09:22:24 +0100 Subject: [PATCH] [postprocessor] Better warning if provided binning method is wrong --- postprocessor.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/postprocessor.py b/postprocessor.py index 6926b53..d0f7520 100644 --- a/postprocessor.py +++ b/postprocessor.py @@ -942,11 +942,16 @@ class PostProcessor(HDF5Container): nb_bin = self.pp_params.disk.nb_bin # radial bins - if self.pp_params.disk.binning == "log": + if self.pp_params.disk.binning in ["log", "logarithmic"]: lrad_in = np.log10(bin_in) lrad_ext = np.log10(bin_out) rad_bins = np.logspace(lrad_in, lrad_ext, num=nb_bin) - elif self.pp_params.disk.binning == "lin": + elif self.pp_params.disk.binning in ["lin", "linear"]: + rad_bins = np.linspace(bin_in, bin_out, num=nb_bin) + else: + raise RuntimeWarning( + f"Invalid parameter {self.pp_params.disk.binning} for disk binning method, using linear binning" + ) rad_bins = np.linspace(bin_in, bin_out, num=nb_bin) # Add boundaries