Merge pull request #34 from offby1/master

ExceptionPexpect.__init__: invoke super's init
pexpect · Jan 7, 2014 · 0b99c3a · 0b99c3a
2 parents f477bc9 + a541386
commit 0b99c3a
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/pexpect/__init__.py b/pexpect/__init__.py
@@ -100,6 +100,7 @@ class ExceptionPexpect(Exception):
     '''
 
     def __init__(self, value):
+        super(ExceptionPexpect, self).__init__(value)
         self.value = value
 
     def __str__(self):

diff --git a/tests/test_pickling.py b/tests/test_pickling.py
@@ -0,0 +1,14 @@
+#!/usr/bin/env python
+import pickle
+import unittest
+
+from pexpect import ExceptionPexpect
+
+class PickleTest(unittest.TestCase):
+    def test_picking(self):
+        e = ExceptionPexpect('Oh noes!')
+        clone = pickle.loads(pickle.dumps(e))
+        self.assertEqual(e.value, clone.value)
+
+if __name__ == '__main__':
+    unittest.main()