Master tests for FireWorks - generally used to ensure that installation was completed properly.
Bases: unittest.case.TestCase
import os import pickle import shutil from unittest import TestCase import unittest import datetime from fireworks import LaunchPad, FireWork, FWorker from fireworks.core.firework import Workflow from fireworks.features.multi_launcher import launch_multiprocess from fireworks.user_objects.firetasks.script_task import ScriptTask
__author__ = ‘xiaohuiqu’
cur_dir = os.getcwd() scr_dir = “launch_scratch” if not os.path.isdir(scr_dir):
os.makedirs(scr_dir)
os.chdir(scr_dir) lp = LaunchPad() lp.reset(password=None, require_password=False) lp.add_wf(FireWork(ScriptTask.from_str(
shell_cmd=’echo “hello 1”’, parameters={“stdout_file”: “task.out”}), fw_id=1))
launch_multiprocess(lp, FWorker(), ‘DEBUG’, 0, 2, 10) fw1 = lp.get_fw_by_id(1) fw2 = lp.get_fw_by_id(2) self.assertEqual(fw1.launches[0].state_history[-1][“state”],
“COMPLETED”)
self.assertEqual(task1out, [‘hello 1
self.assertEqual(task2out, [‘hello 2
TODO: Modify module doc.
Bases: fireworks.core.firework.FireTaskBase