diff options
Diffstat (limited to 'test/helper.py')
| -rw-r--r-- | test/helper.py | 10 | 
1 files changed, 10 insertions, 0 deletions
diff --git a/test/helper.py b/test/helper.py index 2fa45631a..fb8618120 100644 --- a/test/helper.py +++ b/test/helper.py @@ -171,3 +171,13 @@ def assertGreaterEqual(self, got, expected, msg=None):          if msg is None:              msg = '%r not greater than or equal to %r' % (got, expected)          self.assertTrue(got >= expected, msg) + + +def expect_warnings(ydl, warnings_re): +    real_warning = ydl.report_warning + +    def _report_warning(w): +        if not any(re.search(w_re, w) for w_re in warnings_re): +            real_warning(w) + +    ydl.report_warning = _report_warning  | 
