Improve assert_warning() error message for warn_class=None