ip4_networks.py

import ipaddress

import hypothesis.strategies as S


@S.compound
def ip4_networks(draw):
    prefixlen = draw(S.integers(min_value=0, max_value=32))
    if prefixlen:
        network_bits = draw(S.integers(min_value=0, max_value=(1 << prefixlen) - 1))
    else:
        network_bits = 0
    network = network_bits << (32 - prefixlen)
    return ipaddress.IPv4Network((network, prefixlen))