11#!/usr/bin/env python
22# -*- coding: utf-8 -*-
33
4+ import copy
5+ import ipaddress
46import sys
57
68sys .path .append (".." )
79
8- import copy
910import geoip2
1011import requests_mock
1112from geoip2 .errors import (
1819 PermissionRequiredError ,
1920)
2021from geoip2 .webservice import Client
21- from geoip2 .compat import compat_ip_network
2222
2323if sys .version_info [:2 ] == (2 , 6 ):
2424 import unittest2 as unittest
@@ -106,7 +106,7 @@ def test_country_ok(self, mock):
106106 "registered_country is_in_european_union is True" ,
107107 )
108108 self .assertEqual (
109- country .traits .network , compat_ip_network ("1.2.3.0/24" ), "network"
109+ country .traits .network , ipaddress . ip_network ("1.2.3.0/24" ), "network"
110110 )
111111 self .assertEqual (country .raw , self .country , "raw response is correct" )
112112
@@ -315,7 +315,7 @@ def test_city_ok(self, mock):
315315 city = self .client .city ("1.2.3.4" )
316316 self .assertEqual (type (city ), geoip2 .models .City , "return value of client.city" )
317317 self .assertEqual (
318- city .traits .network , compat_ip_network ("1.2.3.0/24" ), "network"
318+ city .traits .network , ipaddress . ip_network ("1.2.3.0/24" ), "network"
319319 )
320320
321321 @requests_mock .mock ()
@@ -331,7 +331,7 @@ def test_insights_ok(self, mock):
331331 type (insights ), geoip2 .models .Insights , "return value of client.insights"
332332 )
333333 self .assertEqual (
334- insights .traits .network , compat_ip_network ("1.2.3.0/24" ), "network"
334+ insights .traits .network , ipaddress . ip_network ("1.2.3.0/24" ), "network"
335335 )
336336 self .assertEqual (insights .traits .static_ip_score , 1.3 , "static_ip_score is 1.3" )
337337 self .assertEqual (insights .traits .user_count , 2 , "user_count is 2" )
0 commit comments