ENH(py): Increase warning stacklevels for tracking

MikeSWang · Sep 19, 2024 · a5211d5 · a5211d5
1 parent afeebca
commit a5211d5
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 10 deletions.
diff --git a/src/triumvirate/_arrayops.py b/src/triumvirate/_arrayops.py
@@ -350,7 +350,8 @@ def extrap_loglin_oscil(a, n_ext):
     if len(a) < 8:
         warnings.warn(
             "Array length is less than 8. Extrapolation may not be accurate.",
-            UserWarning
+            UserWarning,
+            stacklevel=3,
         )
 
     # Check for sign changes and restrict extrapolation to a range
@@ -407,7 +408,8 @@ def extrap_loglin_oscil(a, n_ext):
             warnings.warn(
                 "Divergent behaviour detected at the left tail. "
                 "Constant padding is used instead.",
-                DivergenceWarning
+                DivergenceWarning,
+                stacklevel=3,
             )
             # print("Left tail extrap: trend -> const")
     # No oscillations
@@ -435,7 +437,8 @@ def extrap_loglin_oscil(a, n_ext):
                 warnings.warn(
                     "Divergent behaviour detected at the left endpoints. "
                     "Constant padding is used instead.",
-                    DivergenceWarning
+                    DivergenceWarning,
+                    stacklevel=3,
                 )
                 # print("Left tail extrap: endpoint -> const")
 
@@ -449,7 +452,8 @@ def extrap_loglin_oscil(a, n_ext):
             warnings.warn(
                 "Divergent behaviour detected at the right tail. "
                 "Constant padding is used instead.",
-                DivergenceWarning
+                DivergenceWarning,
+                stacklevel=3,
             )
             # print("Right tail extrap: trend -> const")
     else:
@@ -472,7 +476,8 @@ def extrap_loglin_oscil(a, n_ext):
                 warnings.warn(
                     "Divergent behaviour detected at the right endpoints. "
                     "Constant padding is used instead.",
-                    DivergenceWarning
+                    DivergenceWarning,
+                    stacklevel=3,
                 )
                 # print("Right tail extrap: endpoint -> const")
 

diff --git a/src/triumvirate/threept.py b/src/triumvirate/threept.py
@@ -186,7 +186,8 @@ def _amalgamate_parameters(paramset=None, params_sampling=None,
         warnings.warn(
             "The following parameter default values "
             f"are unchanged: {defaults}. "
-            "Not all parameters are necessarily used."
+            "Not all parameters are necessarily used.",
+            stacklevel=3,
         )
 
     return paramset

diff --git a/src/triumvirate/twopt.py b/src/triumvirate/twopt.py
@@ -148,7 +148,8 @@ def _amalgamate_parameters(paramset=None, params_sampling=None,
         warnings.warn(
             "The following parameter default values "
             f"are unchanged: {defaults}. "
-            "Not all parameters are necessarily used."
+            "Not all parameters are necessarily used.",
+            stacklevel=3,
         )
 
     return paramset

diff --git a/src/triumvirate/winconv.py b/src/triumvirate/winconv.py
@@ -547,7 +547,8 @@ def __init__(self, multipole_Q, multipole_Z, coeff):
         if coeff == 0:
             warnings.warn(
                 "Coefficient of window convolution term is zero.",
-                RuntimeWarning
+                RuntimeWarning,
+                stacklevel=4,
             )
         self.coeff = coeff
 
@@ -1739,7 +1740,8 @@ def initialise_sampts(self, r_in, r_out=None, enforce_coverage=False):
                             "fully covered by the window function "
                             "separation sample points. "
                             "Inaccurate extrapolation may occur. " + msg,
-                            category=ConvolutionRangeWarning
+                            category=ConvolutionRangeWarning,
+                            stacklevel=2,
                         )
                 if (
                     term.multipole_Z not in WinConvTerm._SPEC_INDICES
@@ -1764,7 +1766,8 @@ def initialise_sampts(self, r_in, r_out=None, enforce_coverage=False):
                             "fully covered by the input CF "
                             "separation sample points. "
                             "Inaccurate extrapolation may occur. " + msg,
-                            category=ConvolutionRangeWarning
+                            category=ConvolutionRangeWarning,
+                            stacklevel=2,
                         )