X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=test%2Fvpp_ip.py;h=8b7ea222a67a841dbcee046c6a3c07bdb2bb4930;hb=a462c070ab9bf593d7ed3c454ea3267929550a62;hp=fe985fb901cee7a772bbc5222e1e2493713c1797;hpb=a3aaa61e2f2fd81f9653cb678b38519e96e6c6cd;p=vpp.git

diff --git a/test/vpp_ip.py b/test/vpp_ip.py
index fe985fb901c..8b7ea222a67 100644
--- a/test/vpp_ip.py
+++ b/test/vpp_ip.py
@@ -7,6 +7,10 @@ import logging
 from ipaddress import ip_address
 from socket import AF_INET, AF_INET6
 from vpp_papi import VppEnum
+try:
+    text_type = unicode
+except NameError:
+    text_type = str
 
 _log = logging.getLogger(__name__)
 
@@ -26,7 +30,7 @@ INVALID_INDEX = 0xffffffff
 class VppIpAddressUnion():
     def __init__(self, addr):
         self.addr = addr
-        self.ip_addr = ip_address(unicode(self.addr))
+        self.ip_addr = ip_address(text_type(self.addr))
 
     def encode(self):
         if self.version == 6:
@@ -191,8 +195,8 @@ class VppIpMPrefix():
         self.saddr = saddr
         self.gaddr = gaddr
         self.len = len
-        self.ip_saddr = ip_address(unicode(self.saddr))
-        self.ip_gaddr = ip_address(unicode(self.gaddr))
+        self.ip_saddr = ip_address(text_type(self.saddr))
+        self.ip_gaddr = ip_address(text_type(self.gaddr))
         if self.ip_saddr.version != self.ip_gaddr.version:
             raise ValueError('Source and group addresses must be of the '
                              'same address family.')