iptogeo/tests/test_ip_to_geo.py

#!/usr/bin/env python
# -*- coding: utf-8

import os
from iptogeo import IPToGeo, IPToGeoException

iptogeo = IPToGeo()

def test_ip(ip):
    try:
        (ipres, country_code) = iptogeo.ip_to_geo(ip)
    except IPToGeoException, e:
        print e
        return

    if not country_code:
        print 'Country code for %s (%08x) not found' % (ip, ipres)
    else:
        print 'Country code for %s (%08x) is %s' % (ip, ipres, country_code)

test_ip('1.5.7.3')
test_ip('1.5.255.4')
test_ip('1.6.255.4')
test_ip('2.0.0.0')
test_ip('127.0.0.1')
test_ip('1.55.3.12')
test_ip('1.57.0.0')


for i in range(0, 1000):
    ip = '%d.%d.%d.%d'  % \
        (ord(os.urandom(1)), ord(os.urandom(1)), ord(os.urandom(1)), ord(os.urandom(1)))
    test_ip(ip)