diff options
-rw-r--r-- | netaddr/tests/ip/test_ip.py | 8 | ||||
-rw-r--r-- | netaddr/tests/ip/test_ip_sets.py | 5 |
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()) |