diff options
-rwxr-xr-x | ipaddr_test.py | 11 |
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() |