X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=test%2Fremote_test.py;h=9c825ccf755e0de1f17c8458864e2c0cc78534a9;hb=053204ab039d34a990ff0e14c32ce3b294fcce0e;hp=cd2e46fe219e61a41c9515405144d309c8b72119;hpb=df36f4963f1a590ce9a02f048507c3d4590580ae;p=vpp.git

diff --git a/test/remote_test.py b/test/remote_test.py
index cd2e46fe219..9c825ccf755 100644
--- a/test/remote_test.py
+++ b/test/remote_test.py
@@ -3,14 +3,13 @@
 import inspect
 import os
 import unittest
+from framework import VppTestCase
 from multiprocessing import Process, Pipe
 from pickle import dumps
-
 import six
 from six import moves
-
-from framework import VppTestCase
-from aenum import Enum
+import sys
+from aenum import IntEnum, IntFlag
 
 
 class SerializableClassCopy(object):
@@ -270,7 +269,7 @@ class RemoteClass(Process):
             if type(obj) is tuple:
                 rv = tuple(rv)
             return rv
-        elif (isinstance(obj, Enum)):
+        elif (isinstance(obj, IntEnum) or isinstance(obj, IntFlag)):
             return obj.value
         else:
             return self._make_obj_serializable(obj)