[WIP] Update warning stacklevel when wrapping warnings.warn

2018-10-18 02:36:41 +00:00
parent 7931b5b489
commit 1b51497936
2 changed files with 8 additions and 0 deletions
--- a/src/_pytest/recwarn.py
+++ b/src/_pytest/recwarn.py
@@ -156,6 +156,8 @@ class WarningsRecorder(warnings.catch_warnings):
        if six.PY2:

            def warn(*args, **kwargs):
+                kwargs.setdefault("stacklevel", 1)
+                kwargs["stacklevel"] += 1
                return self._saved_warn(*args, **kwargs)

            warnings.warn, self._saved_warn = warn, warnings.warn