Similar to assertRaises and assertRaisesRegexp, unittest should provide assertWarns and assertWarnsRegexp, to check that a given callable (or piece of code) triggers a particular warning.

Currently, you have to do that manually using a mixture of warnings.catch_warnings and warnings.filterwarnings, which is pretty annoying.

+1

The __warningregistry__ stuff looks horrible.

> unittest should provide assertWarns and assertWarnsRegexp

+1
(the internal helpers in test.support could be refactored)

> The __warningregistry__ stuff looks horrible.

+1
(and it does not behave exactly the same in all Python versions)

+1 from me as well.

Here is a patch. The approach is different from support.check_warnings(), and tries to mimic assertRaises* instead.

There was some dead code in the patch.

(Note that in general I am against extending the TestCase API with more asserts given how wide it is and how much it has expanded in recent versions. I've written warning checking code enough times for third party projects that I think this is worth it though.)

Patch looks good. However tests do not pass with -Werror (while test_warnings and others pass).


Is there a way to catch multiple warnings on a single logical line?
(With assertRaises we don't have such use case)

> Patch looks good. However tests do not pass with -Werror (while test_warnings and others pass).
> 
> 
> Is there a way to catch multiple warnings on a single logical line?

I thought we could beef up the API with additional arguments.
That would be at the expense of removing the "callable" argument (and
mandating the use as a context manager), though, otherwise the signature
gets too complicated.

Why not accepting a tuple of warnings? That doesn't make sense for assertWarnsRegexp of course.

> Why not accepting a tuple of warnings?

It already does (and there's a test!).

> That doesn't make sense for assertWarnsRegexp of course.

True.

Updated patch so that the tests pass with -Werror.
Do you think this should be committed before the next alpha?

Committed in r84563!