def test_ip_afi(): assert IP.get_afi("192.168.0.0/24") == "4" assert IP.get_afi("::1/128") == "6"
def test_ip_afi(prefix, afi): assert IP.get_afi(prefix) == afi