tests: python3 version check for enum module

[vpp.git] / test / remote_test.py

diff --git a/test/remote_test.py b/test/remote_test.py
index 1b42f8a..d084133 100644 (file)
--- a/test/remote_test.py
+++ b/test/remote_test.py
@@ -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):