def test_cache_unique_id_false(self):
_DummyTaskDep.param = luigi.Parameter(default='original_param')
output1 = gokart.build(_DummyTask(cache_unique_id=False),
reset_register=False)
_DummyTaskDep.param = luigi.Parameter(default='updated_param')
output2 = gokart.build(_DummyTask(cache_unique_id=False),
reset_register=False)
self.assertNotEqual(output1, output2)
def test_read_config(self):
os.environ.setdefault('test_param', 'test')
config_file_path = os.path.join(
os.path.dirname(os.path.abspath(__file__)), 'config',
'test_config.ini')
gokart.utils.add_config(config_file_path)
output = gokart.build(_DummyTask(), reset_register=False)
self.assertEqual(output, 'test')
def test_build_dict_outputs(self):
param_dict = {
'out1': 'test1',
'out2': 'test2',
}
output = gokart.build(_DummyTaskTwoOutputs(param1=param_dict['out1'],
param2=param_dict['out2']),
reset_register=False)
self.assertEqual(output, param_dict)
def test_failed_task(self):
with self.assertRaises(GokartBuildError):
gokart.build(_DummyFailedTask(), reset_register=False)
def test_build(self):
text = 'test'
output = gokart.build(_DummyTask(param=text), reset_register=False)
self.assertEqual(output, text)
def test_build_parallel(self):
output = gokart.build(_ParallelRunner(),
reset_register=False,
workers=20)
self.assertEqual(output, 'done')
