aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/remote_test.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/test/remote_test.py b/test/remote_test.py
index 1b42f8a0189..d084133d475 100644
--- 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):