VPP-1508 python tests: unicode

[vpp.git] / test / test_ip_ecmp.py

diff --git a/test/test_ip_ecmp.py b/test/test_ip_ecmp.py
index 60673ea..17b0a6c 100644 (file)
--- a/test/test_ip_ecmp.py
+++ b/test/test_ip_ecmp.py
@@ -5,6 +5,7 @@ import random
 import socket
 from ipaddress import IPv4Address, IPv6Address, AddressValueError
 
+
 from framework import VppTestCase, VppTestRunner
 from util import ppp
 
@@ -13,6 +14,11 @@ from scapy.layers.l2 import Ether
 from scapy.layers.inet import IP, UDP
 from scapy.layers.inet6 import IPv6
 
+try:
+    text_type = unicode
+except NameError:
+    text_type = str
+
 #
 # The number of packets to sent.
 #
@@ -79,10 +85,10 @@ class TestECMP(VppTestCase):
         :return: Random IPv4 or IPv6 address from required range.
         """
         try:
-            ip_addr = IPv4Address(unicode(ip_addr_start))
+            ip_addr = IPv4Address(text_type(ip_addr_start))
             ip_max_len = 32
         except (AttributeError, AddressValueError):
-            ip_addr = IPv6Address(unicode(ip_addr_start))
+            ip_addr = IPv6Address(text_type(ip_addr_start))
             ip_max_len = 128
 
         return str(ip_addr +