1 files changed, 28 insertions, 30 deletions
diff --git a/sources/pyside2/tests/signals/signal_enum_test.py b/sources/pyside2/tests/signals/signal_enum_test.py
index 3a24b919d..d48aa10cf 100644
--- a/sources/pyside2/tests/signals/signal_enum_test.py
+++ b/sources/pyside2/tests/signals/signal_enum_test.py
@@ -26,6 +26,7 @@
 ##
 #############################################################################
 
+from enum import Enum
 import os
 import sys
 import unittest
@@ -36,47 +37,44 @@ from init_paths import init_test_paths
 init_test_paths(False)
 
 from PySide2.QtCore import QObject, Signal, Slot
-import py3kcompat as py3k
 
 
-if py3k.IS_PY3K:
-    from enum import Enum
+class Colors(Enum):
+    red = 1
+    green = 2
+    blue = 3
 
-    class Colors(Enum):
-        red = 1
-        green = 2
-        blue = 3
+class Obj(QObject):
+    enum_signal = Signal(Colors)
+    object_signal = Signal(object)
 
-    class Obj(QObject):
-        enum_signal = Signal(Colors)
-        object_signal = Signal(object)
+    def __init__(self, parent=None):
+        QObject.__init__(self, parent)
+        self.enum_signal.connect(self.get_result)
+        self.object_signal.connect(self.get_result)
+        self.value = -1
 
-        def __init__(self, parent=None):
-            QObject.__init__(self, parent)
-            self.enum_signal.connect(self.get_result)
-            self.object_signal.connect(self.get_result)
-            self.value = -1
+    @Slot()
+    def get_result(self, i):
+        self.value = i
 
-        @Slot()
-        def get_result(self, i):
-            self.value = i
 
+class SignalEnumTests(unittest.TestCase):
+    '''Test Signal with enum.Enum'''
 
-    class SignalEnumTests(unittest.TestCase):
-        '''Test Signal with enum.Enum'''
+    def testSignal(self):
+        o = Obj()
+        # Default value
+        self.assertEqual(o.value, -1)
 
-        def testSignal(self):
-            o = Obj()
-            # Default value
-            self.assertEqual(o.value, -1)
+        # Enum Signal
+        o.enum_signal.emit(Colors.green)
+        self.assertEqual(o.value, Colors.green)
 
-            # Enum Signal
-            o.enum_signal.emit(Colors.green)
-            self.assertEqual(o.value, Colors.green)
+        # object Signal
+        o.object_signal.emit(Colors.red)
+        self.assertEqual(o.value, Colors.red)
 
-            # object Signal
-            o.object_signal.emit(Colors.red)
-            self.assertEqual(o.value, Colors.red)
 
 if __name__ == '__main__':
     unittest.main()