Python output Examples

Example #1

 def test_run_with_failure(self):
     task = self.create_task()
     task.output().touch = MagicMock(side_effect=Exception("Failed to update marker"))
     with self.assertRaises(Exception):
         task.run()
     self.assertTrue(self.mock_mysql_connector.connect().cursor().execute.called)
     self.assertTrue(self.mock_mysql_connector.connect().rollback.called)
     self.assertFalse(self.mock_mysql_connector.connect().commit.called)
     self.assertTrue(self.mock_mysql_connector.connect().close.called)

Example #2

File: test_mysql_load.py Project: TsukishirOSan/edx-analytics-pipeline

 def test_run_with_failure(self):
     task = self.create_task()
     task.output().touch = MagicMock(side_effect=Exception("Failed to update marker"))
     with self.assertRaises(Exception):
         task.run()
     self.assertTrue(self.mock_mysql_connector.connect().cursor().execute.called)
     self.assertTrue(self.mock_mysql_connector.connect().rollback.called)
     self.assertFalse(self.mock_mysql_connector.connect().commit.called)
     self.assertTrue(self.mock_mysql_connector.connect().close.called)

Example #3

 def output(self):
     return [task.output() for task in self.requires()]

Example #4

File: module_engagement.py Project: DucQuang1/edx-analytics-pipeline

 def output(self):
     return [task.output() for task in self.requires()]

Example #5

def remove_targets(task):
    outputs = luigi.task.flatten(task.output())
    for out in outputs:
        if out.exists():
            out.remove()
        assert not out.exists()