Verify that various objects are weak referencable

2 files changed, 13 insertions, 0 deletions

diff --git a/netaddr/tests/ip/test_ip.py b/netaddr/tests/ip/test_ip.py
new file mode 100644
index 0000000..3ffa83f
--- /dev/null
+++ b/netaddr/tests/ip/test_ip.py
@@ -0,0 +1,8 @@
+import weakref
+
+from netaddr import IPAddress, IPNetwork, IPRange
+
+def test_ip_classes_are_weak_referencable():
+    weakref.ref(IPAddress('10.0.0.1'))
+    weakref.ref(IPNetwork('10.0.0.1/8'))
+    weakref.ref(IPRange('10.0.0.1', '10.0.0.10'))
diff --git a/netaddr/tests/ip/test_ip_sets.py b/netaddr/tests/ip/test_ip_sets.py
index 0a3391b..d7fdc4f 100644
--- a/netaddr/tests/ip/test_ip_sets.py
+++ b/netaddr/tests/ip/test_ip_sets.py
@@ -1,4 +1,5 @@
 import pickle
+import weakref
 
 import pytest
 
@@ -574,3 +575,7 @@ def test_ipset_converts_to_cidr_networks_v6():
         IPNetwork('fe80::/64'),
         IPNetwork('fe90::/64'),
     ]
+
+
+def test_ipset_is_weak_referencable():
+    weakref.ref(IPSet())