tests: python3 version check for enum module

Type: fix
Change-Id: Ie210500fba741d3f047e8499957b5b233a78fa36
Signed-off-by: snaramre <snaramre@cisco.com>

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):