class IpLocatorTest(unittest.TestCase): def setUp(self): self.ip_locator = IpLocator() def test_locate_using_GeoIP(self): addr = '94.1.203.100' expected_result = (0.49111111097865634, 0.18916666242811417, u'Edinburgh') self.assertEqual(expected_result, self.ip_locator.locate(addr)) def test_locate_using_db_extension(self): addr = '81.234.35.62' expected_result = (0.5332408333333334, 0.17935166666666666, u'Göteborg') self.assertEqual(expected_result, self.ip_locator.locate(addr))
def setUp(self): self.ip_locator = IpLocator()
#!/usr/bin/python import sys import re import cPickle sys.path.append("..") from ip_locator import IpLocator ip_locator = IpLocator() locations = [] f = open("peers.log") r = re.compile('^peer \[([0-9.]+)\]') for line in f: m = r.search(line) if m: addr = m.group(1) location = ip_locator.locate(addr) if location: locations.append(location) f.close() out = open("locations.dat", "w") cPickle.dump(locations, out) out.close()