diff options
Diffstat (limited to 'Lib/warnings.py')
-rw-r--r-- | Lib/warnings.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/Lib/warnings.py b/Lib/warnings.py index 70d087e..16246b4 100644 --- a/Lib/warnings.py +++ b/Lib/warnings.py @@ -169,7 +169,9 @@ def warn(message, category=None, stacklevel=1): # Check category argument if category is None: category = UserWarning - assert issubclass(category, Warning) + if not (isinstance(category, type) and issubclass(category, Warning)): + raise TypeError("category must be a Warning subclass, " + "not '{:s}'".format(type(category).__name__)) # Get context information try: caller = sys._getframe(stacklevel) @@ -186,7 +188,7 @@ def warn(message, category=None, stacklevel=1): filename = globals.get('__file__') if filename: fnl = filename.lower() - if fnl.endswith((".pyc", ".pyo")): + if fnl.endswith(".pyc"): filename = filename[:-1] else: if module == "__main__": |