diff --git a/boxtree/distributed/__init__.py b/boxtree/distributed/__init__.py index b2c9d19d..4883c496 100644 --- a/boxtree/distributed/__init__.py +++ b/boxtree/distributed/__init__.py @@ -217,7 +217,8 @@ def make_distributed_wrangler( # TODO: should replace the default calibration params with a more # accurate one warnings.warn("Calibration parameters for the cost model are not " - "supplied. The default one will be used.") + "supplied. The default one will be used.", + stacklevel=2) calibration_params = \ FMMCostModel.get_unit_calibration_params() diff --git a/boxtree/tree.py b/boxtree/tree.py index 76d620e1..3115bee4 100644 --- a/boxtree/tree.py +++ b/boxtree/tree.py @@ -1191,7 +1191,7 @@ def filter_target_lists_in_user_order(queue, tree, flags): "filter_target_lists_in_user_order() is deprecated and will go " "away in a future release. Use " "ParticleListFilter.filter_target_lists_in_user_order() instead.", - DeprecationWarning) + DeprecationWarning, stacklevel=2) return (ParticleListFilter(queue.context) .filter_target_lists_in_user_order(queue, tree, flags)) @@ -1206,7 +1206,7 @@ def filter_target_lists_in_tree_order(queue, tree, flags): "filter_target_lists_in_tree_order() is deprecated and will go " "away in a future release. Use " "ParticleListFilter.filter_target_lists_in_tree_order() instead.", - DeprecationWarning) + DeprecationWarning, stacklevel=2) return (ParticleListFilter(queue.context) .filter_target_lists_in_tree_order(queue, tree, flags)) diff --git a/boxtree/tree_of_boxes.py b/boxtree/tree_of_boxes.py index 7ca96425..425466af 100644 --- a/boxtree/tree_of_boxes.py +++ b/boxtree/tree_of_boxes.py @@ -223,7 +223,7 @@ def _apply_coarsen_flags(coarsen_flags, tob, error_on_ignored_flags=True): raise RuntimeError(msg) else: import warnings - warnings.warn(msg) + warnings.warn(msg, stacklevel=3) # deleted boxes are marked as: # level = inf diff --git a/test/test_fmm.py b/test/test_fmm.py index 28c2f359..f66af872 100644 --- a/test/test_fmm.py +++ b/test/test_fmm.py @@ -482,7 +482,7 @@ def fmm_level_to_order(tree, lev): have_sumpy = False from warnings import warn warn("sumpy unavailable: cannot compute independent reference " - "values for pyfmmlib") + "values for pyfmmlib", stacklevel=1) else: have_sumpy = True