[test][worker] Code cleanup

beat/core/test/test_worker.py

@@ -37,19 +37,12 @@
 # Tests for experiment execution
 
 import os
-
 import logging
-
-logger = logging.getLogger(__name__)
-
 import unittest
 import simplejson
 import multiprocessing
+import queue
 
-try:
-    import Queue
-except ImportError:
-    import queue as Queue
 from time import time
 from time import sleep
 
@@ -60,6 +53,9 @@ from ..utils import find_free_port
 
 from . import prefix, tmp_prefix
 
+
+logger = logging.getLogger(__name__)
+
 # ----------------------------------------------------------
 
 
@@ -165,7 +161,7 @@ class ControllerProcess(multiprocessing.Process):
                 command = self.queue.get_nowait()
                 if command == "STOP":
                     break
-            except Queue.Empty:
+            except queue.Empty:
                 pass
 
         self.controller.destroy()
@@ -390,7 +386,7 @@ class TestOneWorker(TestWorkerBase):
         return message
 
     def _check_done(self, message, expected_worker, expected_job_id):
-        self.assertTrue(message is not None)
+        self.assertIsNotNone(message)
 
         (worker, status, job_id, data) = message