nat: use correct data types for memory sizes

[vpp.git] / test / remote_test.py

diff --git a/test/remote_test.py b/test/remote_test.py
index 1b42f8a..8b3def2 100644 (file)
--- a/test/remote_test.py
+++ b/test/remote_test.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
 import inspect
 import os
@@ -9,7 +9,11 @@ from pickle import dumps
 import six
 from six import moves
 import sys
-from aenum import IntEnum, IntFlag
+
+if sys.version_info < (3,):
+    from aenum import IntEnum, IntFlag
+else:
+    from enum import IntEnum, IntFlag
 
 
 class SerializableClassCopy(object):