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")