contextlib.nested was removed in python 3, using native syntax instead

spotify · Nov 29, 2015 · 05a89de · 05a89de
1 parent 69a5a63
commit 05a89de
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/test/worker_test.py b/test/worker_test.py
@@ -25,7 +25,6 @@
 import threading
 import time
 import psutil
-import contextlib
 from helpers import unittest, with_config, skipOnTravis, LuigiTestCase
 
 import luigi.notifications
@@ -113,8 +112,8 @@ class WorkerTest(unittest.TestCase):
     def run(self, result=None):
         self.sch = CentralPlannerScheduler(retry_delay=100, remove_delay=1000, worker_disconnect_delay=10)
         self.time = time.time
-        with contextlib.nested(Worker(scheduler=self.sch, worker_id='X'),
-                               Worker(scheduler=self.sch, worker_id='Y')) as (w, w2):
+        with Worker(scheduler=self.sch, worker_id='X') as w, \
+             Worker(scheduler=self.sch, worker_id='Y') as w2:
             self.w = w
             self.w2 = w2
             super(WorkerTest, self).run(result)