From 5b72d1d867b7bcaf1276559098a76049506f9168 Mon Sep 17 00:00:00 2001 From: Michael Shields Date: Wed, 2 Sep 2015 15:07:36 -0700 Subject: Test pickling. Closes #106. --- ipaddr_test.py | 11 +++++++++-- 1 file 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() -- cgit v1.2.1