Base class for all errors

Closes #43
stub42 · Apr 28, 2020 · 238b4f6 · 238b4f6
1 parent fb8b146
commit 238b4f6
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/pytz/exceptions.py b/src/pytz/exceptions.py
@@ -8,7 +8,11 @@
 ]
 
 
-class UnknownTimeZoneError(KeyError):
+class Error(Exception):
+    '''Base class for all exceptions raised by the pytz library'''
+
+
+class UnknownTimeZoneError(KeyError, Error):
     '''Exception raised when pytz is passed an unknown timezone.
 
     >>> isinstance(UnknownTimeZoneError(), LookupError)
@@ -20,11 +24,18 @@ class UnknownTimeZoneError(KeyError):
 
     >>> isinstance(UnknownTimeZoneError(), KeyError)
     True
+
+    And also a subclass of pytz.exceptions.Error, as are other pytz
+    exceptions.
+
+    >>> isinstance(UnknownTimeZoneError(), Error)
+    True
+
     '''
     pass
 
 
-class InvalidTimeError(Exception):
+class InvalidTimeError(Error):
     '''Base class for invalid time exceptions.'''
 
 

diff --git a/src/pytz/tests/test_tzinfo.py b/src/pytz/tests/test_tzinfo.py
@@ -859,6 +859,7 @@ def test_suite():
     suite = unittest.TestSuite()
     suite.addTest(doctest.DocTestSuite('pytz'))
     suite.addTest(doctest.DocTestSuite('pytz.tzinfo'))
+    suite.addTest(doctest.DocTestSuite('pytz.exceptions'))
     import test_tzinfo
     suite.addTest(unittest.defaultTestLoader.loadTestsFromModule(test_tzinfo))
     return suite