summaryrefslogtreecommitdiffstats
path: root/test/test_ip_ecmp.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_ip_ecmp.py')
-rw-r--r--test/test_ip_ecmp.py10
1 files changed, 8 insertions, 2 deletions
diff --git a/test/test_ip_ecmp.py b/test/test_ip_ecmp.py
index 60673ea6f3a..17b0a6c79c2 100644
--- 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 +