Test pickling.

Closes #106.

1 files changed, 9 insertions, 2 deletions

diff --git a/ipaddr_test.py b/ipaddr_test.py
index 79a18c4..30b4ccb 100755
--- a/ipaddr_test.py
+++ b/ipaddr_test.py
@@ -17,9 +17,10 @@
 
 """Unittest for ipaddr module."""
 
-
-import unittest
+import pickle
 import time
+import unittest
+
 import ipaddr
 
 # Compatibility function to cast str to bytes objects
@@ -1216,6 +1217,12 @@ class IpaddrUnitTest(unittest.TestCase):
                          sixtofouraddr.sixtofour)
         self.assertFalse(bad_addr.sixtofour)
 
+    def testPickle(self):
+        for obj in (
+            ipaddr.IPv4Address('192.0.2.1'), ipaddr.IPv6Address('2001:db8::1'),
+            ipaddr.IPv4Network('192.0.2.1'), ipaddr.IPv6Network('2001:db8::1')):
+            self.assertEqual(obj, pickle.loads(pickle.dumps(obj)))
+
 
 if __name__ == '__main__':
     unittest.main()