def setUp(self):
self.worker = Worker()
self.worker_scheduler_factory = _WorkerSchedulerFactory()
self.worker_scheduler_factory.create_worker = Mock(return_value=self.worker)
self.worker_scheduler_factory.create_local_scheduler = Mock()
super(InterfaceTest, self).setUp()
class NoOpTask(luigi.Task):
param = luigi.Parameter()
self.task_a = NoOpTask("a")
self.task_b = NoOpTask("b")
def setUp(self):
self.worker = Worker()
self.worker.stop = Mock()
self.worker_scheduler_factory = _WorkerSchedulerFactory()
self.worker_scheduler_factory.create_worker = Mock(return_value=self.worker)
self.worker_scheduler_factory.create_local_scheduler = Mock()
super(InterfaceTest, self).setUp()
class NoOpTask(luigi.Task):
param = luigi.Parameter()
self.task_a = NoOpTask("a")
self.task_b = NoOpTask("b")