fireworks.core.tests package

Submodules

fireworks.core.tests.tasks module

class fireworks.core.tests.tasks.DetoursTask(*args, **kwargs)

Bases: FiretaskBase

optional_params = ['n_detours', 'data_per_detour']
run_task(fw_spec)

This method gets called when the Firetask is run. It can take in a Firework spec, perform some task using that data, and then return an output in the form of a FWAction.

Parameters:

fw_spec (dict) – A Firework spec. This comes from the master spec. In addition, this spec contains a special “_fw_env” key that contains the env settings of the FWorker calling this method. This provides for abstracting out certain commands or settings. For example, “foo” may be named “foo1” in resource 1 and “foo2” in resource 2. The FWorker env can specify { “foo”: “foo1”}, which maps an abstract variable “foo” to the relevant “foo1” or “foo2”. You can then write a task that uses fw_spec[“_fw_env”][“foo”] that will work across all these multiple resources.

Returns:

(FWAction)

class fireworks.core.tests.tasks.DoNothingTask(*args, **kwargs)

Bases: FiretaskBase

run_task(fw_spec) None

This method gets called when the Firetask is run. It can take in a Firework spec, perform some task using that data, and then return an output in the form of a FWAction.

Parameters:

fw_spec (dict) – A Firework spec. This comes from the master spec. In addition, this spec contains a special “_fw_env” key that contains the env settings of the FWorker calling this method. This provides for abstracting out certain commands or settings. For example, “foo” may be named “foo1” in resource 1 and “foo2” in resource 2. The FWorker env can specify { “foo”: “foo1”}, which maps an abstract variable “foo” to the relevant “foo1” or “foo2”. You can then write a task that uses fw_spec[“_fw_env”][“foo”] that will work across all these multiple resources.

Returns:

(FWAction)

class fireworks.core.tests.tasks.ExceptionTestTask(*args, **kwargs)

Bases: FiretaskBase

exec_counter = 0
run_task(fw_spec) None

This method gets called when the Firetask is run. It can take in a Firework spec, perform some task using that data, and then return an output in the form of a FWAction.

Parameters:

fw_spec (dict) – A Firework spec. This comes from the master spec. In addition, this spec contains a special “_fw_env” key that contains the env settings of the FWorker calling this method. This provides for abstracting out certain commands or settings. For example, “foo” may be named “foo1” in resource 1 and “foo2” in resource 2. The FWorker env can specify { “foo”: “foo1”}, which maps an abstract variable “foo” to the relevant “foo1” or “foo2”. You can then write a task that uses fw_spec[“_fw_env”][“foo”] that will work across all these multiple resources.

Returns:

(FWAction)

class fireworks.core.tests.tasks.ExecutionCounterTask(*args, **kwargs)

Bases: FiretaskBase

exec_counter = 0
run_task(fw_spec) None

This method gets called when the Firetask is run. It can take in a Firework spec, perform some task using that data, and then return an output in the form of a FWAction.

Parameters:

fw_spec (dict) – A Firework spec. This comes from the master spec. In addition, this spec contains a special “_fw_env” key that contains the env settings of the FWorker calling this method. This provides for abstracting out certain commands or settings. For example, “foo” may be named “foo1” in resource 1 and “foo2” in resource 2. The FWorker env can specify { “foo”: “foo1”}, which maps an abstract variable “foo” to the relevant “foo1” or “foo2”. You can then write a task that uses fw_spec[“_fw_env”][“foo”] that will work across all these multiple resources.

Returns:

(FWAction)

class fireworks.core.tests.tasks.MalformedAdditionTask(*args, **kwargs)

Bases: FiretaskBase

run_task(fw_spec)

This method gets called when the Firetask is run. It can take in a Firework spec, perform some task using that data, and then return an output in the form of a FWAction.

Parameters:

fw_spec (dict) – A Firework spec. This comes from the master spec. In addition, this spec contains a special “_fw_env” key that contains the env settings of the FWorker calling this method. This provides for abstracting out certain commands or settings. For example, “foo” may be named “foo1” in resource 1 and “foo2” in resource 2. The FWorker env can specify { “foo”: “foo1”}, which maps an abstract variable “foo” to the relevant “foo1” or “foo2”. You can then write a task that uses fw_spec[“_fw_env”][“foo”] that will work across all these multiple resources.

Returns:

(FWAction)

exception fireworks.core.tests.tasks.SerializableException(exc_details)

Bases: Exception

__init__(exc_details) None
to_dict()
class fireworks.core.tests.tasks.SlowAdditionTask(*args, **kwargs)

Bases: FiretaskBase

run_task(fw_spec)

This method gets called when the Firetask is run. It can take in a Firework spec, perform some task using that data, and then return an output in the form of a FWAction.

Parameters:

fw_spec (dict) – A Firework spec. This comes from the master spec. In addition, this spec contains a special “_fw_env” key that contains the env settings of the FWorker calling this method. This provides for abstracting out certain commands or settings. For example, “foo” may be named “foo1” in resource 1 and “foo2” in resource 2. The FWorker env can specify { “foo”: “foo1”}, which maps an abstract variable “foo” to the relevant “foo1” or “foo2”. You can then write a task that uses fw_spec[“_fw_env”][“foo”] that will work across all these multiple resources.

Returns:

(FWAction)

class fireworks.core.tests.tasks.SlowTodictTask(*args, **kwargs)

Bases: FiretaskBase

run_task(fw_spec)

This method gets called when the Firetask is run. It can take in a Firework spec, perform some task using that data, and then return an output in the form of a FWAction.

Parameters:

fw_spec (dict) – A Firework spec. This comes from the master spec. In addition, this spec contains a special “_fw_env” key that contains the env settings of the FWorker calling this method. This provides for abstracting out certain commands or settings. For example, “foo” may be named “foo1” in resource 1 and “foo2” in resource 2. The FWorker env can specify { “foo”: “foo1”}, which maps an abstract variable “foo” to the relevant “foo1” or “foo2”. You can then write a task that uses fw_spec[“_fw_env”][“foo”] that will work across all these multiple resources.

Returns:

(FWAction)

to_dict()
class fireworks.core.tests.tasks.TodictErrorTask(*args, **kwargs)

Bases: FiretaskBase

run_task(fw_spec)

This method gets called when the Firetask is run. It can take in a Firework spec, perform some task using that data, and then return an output in the form of a FWAction.

Parameters:

fw_spec (dict) – A Firework spec. This comes from the master spec. In addition, this spec contains a special “_fw_env” key that contains the env settings of the FWorker calling this method. This provides for abstracting out certain commands or settings. For example, “foo” may be named “foo1” in resource 1 and “foo2” in resource 2. The FWorker env can specify { “foo”: “foo1”}, which maps an abstract variable “foo” to the relevant “foo1” or “foo2”. You can then write a task that uses fw_spec[“_fw_env”][“foo”] that will work across all these multiple resources.

Returns:

(FWAction)

to_dict() NoReturn
class fireworks.core.tests.tasks.WaitWFLockTask(*args, **kwargs)

Bases: FiretaskBase

run_task(fw_spec)

This method gets called when the Firetask is run. It can take in a Firework spec, perform some task using that data, and then return an output in the form of a FWAction.

Parameters:

fw_spec (dict) – A Firework spec. This comes from the master spec. In addition, this spec contains a special “_fw_env” key that contains the env settings of the FWorker calling this method. This provides for abstracting out certain commands or settings. For example, “foo” may be named “foo1” in resource 1 and “foo2” in resource 2. The FWorker env can specify { “foo”: “foo1”}, which maps an abstract variable “foo” to the relevant “foo1” or “foo2”. You can then write a task that uses fw_spec[“_fw_env”][“foo”] that will work across all these multiple resources.

Returns:

(FWAction)

fireworks.core.tests.test_firework module

TODO: Modify unittest doc.

class fireworks.core.tests.test_firework.FiretaskBaseTest(methodName='runTest')

Bases: TestCase

test_init() None
test_param_checks() None
class fireworks.core.tests.test_firework.FiretaskPickleTest(methodName='runTest')

Bases: TestCase

setUp() None

Hook method for setting up the test fixture before exercising it.

test_init() None
class fireworks.core.tests.test_firework.PickleTask(*args, **kwargs)

Bases: FiretaskBase

required_params = ['test']
run_task(fw_spec)

This method gets called when the Firetask is run. It can take in a Firework spec, perform some task using that data, and then return an output in the form of a FWAction.

Parameters:

fw_spec (dict) – A Firework spec. This comes from the master spec. In addition, this spec contains a special “_fw_env” key that contains the env settings of the FWorker calling this method. This provides for abstracting out certain commands or settings. For example, “foo” may be named “foo1” in resource 1 and “foo2” in resource 2. The FWorker env can specify { “foo”: “foo1”}, which maps an abstract variable “foo” to the relevant “foo1” or “foo2”. You can then write a task that uses fw_spec[“_fw_env”][“foo”] that will work across all these multiple resources.

Returns:

(FWAction)

class fireworks.core.tests.test_firework.Task1(*args, **kwargs)

Bases: FiretaskBase

run_task(fw_spec)

This method gets called when the Firetask is run. It can take in a Firework spec, perform some task using that data, and then return an output in the form of a FWAction.

Parameters:

fw_spec (dict) – A Firework spec. This comes from the master spec. In addition, this spec contains a special “_fw_env” key that contains the env settings of the FWorker calling this method. This provides for abstracting out certain commands or settings. For example, “foo” may be named “foo1” in resource 1 and “foo2” in resource 2. The FWorker env can specify { “foo”: “foo1”}, which maps an abstract variable “foo” to the relevant “foo1” or “foo2”. You can then write a task that uses fw_spec[“_fw_env”][“foo”] that will work across all these multiple resources.

Returns:

(FWAction)

class fireworks.core.tests.test_firework.Task2(*args, **kwargs)

Bases: FiretaskBase

run_task(fw_spec)

This method gets called when the Firetask is run. It can take in a Firework spec, perform some task using that data, and then return an output in the form of a FWAction.

Parameters:

fw_spec (dict) – A Firework spec. This comes from the master spec. In addition, this spec contains a special “_fw_env” key that contains the env settings of the FWorker calling this method. This provides for abstracting out certain commands or settings. For example, “foo” may be named “foo1” in resource 1 and “foo2” in resource 2. The FWorker env can specify { “foo”: “foo1”}, which maps an abstract variable “foo” to the relevant “foo1” or “foo2”. You can then write a task that uses fw_spec[“_fw_env”][“foo”] that will work across all these multiple resources.

Returns:

(FWAction)

class fireworks.core.tests.test_firework.WorkflowTest(methodName='runTest')

Bases: TestCase

setUp() None

Hook method for setting up the test fixture before exercising it.

test_copy() None

Test that we can produce a copy of a Workflow but that the copy has unique fw_ids.

test_init() None
test_iter_len_index() None
test_remove_leaf_fws() None
test_remove_root_fws() None

fireworks.core.tests.test_launchpad module

class fireworks.core.tests.test_launchpad.AuthenticationTest(methodName='runTest')

Bases: TestCase

Tests whether users are authenticating against the correct mongo dbs.

classmethod setUpClass() None

Hook method for setting up class fixture before running tests in the class.

test_authenticating_to_users_db() None

A user should be able to authenticate against a database that they are a user of.

test_authsource_infered_from_db_name() None

The default behavior is to authenticate against the db that the user is trying to access.

test_no_admin_privileges_for_plebs() None

Normal users can not authenticate against the admin db.

class fireworks.core.tests.test_launchpad.GridfsStoredDataTest(methodName='runTest')

Bases: TestCase

Tests concerning the storage of data in Gridfs when the size of the documents exceed the 16MB limit.

pytestmark = [Mark(name='mongodb', args=(), kwargs={})]
setUp() None

Hook method for setting up the test fixture before exercising it.

classmethod setUpClass() None

Hook method for setting up class fixture before running tests in the class.

tearDown() None

Hook method for deconstructing the test fixture after testing it.

classmethod tearDownClass() None

Hook method for deconstructing the class fixture after running all tests in the class.

test_many_detours() None
test_many_detours_offline() None
class fireworks.core.tests.test_launchpad.LaunchPadDefuseReigniteRerunArchiveDeleteTest(methodName='runTest')

Bases: TestCase

setUp() None

Hook method for setting up the test fixture before exercising it.

classmethod setUpClass() None

Hook method for setting up class fixture before running tests in the class.

tearDown() None

Hook method for deconstructing the test fixture after testing it.

classmethod tearDownClass() None

Hook method for deconstructing the class fixture after running all tests in the class.

test_archive_wf() None
test_defuse_fw() None
test_defuse_fw_after_completion() None
test_defuse_wf() None
test_defuse_wf_after_partial_run() None
test_delete_wf() None
test_delete_wf_and_files() None
test_pause_fw() None
test_pause_wf() None
test_reignite_fw() None
test_reignite_wf() None
test_rerun_fws2() None
class fireworks.core.tests.test_launchpad.LaunchPadLostRunsDetectTest(methodName='runTest')

Bases: TestCase

setUp() None

Hook method for setting up the test fixture before exercising it.

classmethod setUpClass() None

Hook method for setting up class fixture before running tests in the class.

tearDown() None

Hook method for deconstructing the test fixture after testing it.

classmethod tearDownClass() None

Hook method for deconstructing the class fixture after running all tests in the class.

test_detect_lostruns() None
test_detect_lostruns_defuse() None
test_state_after_run_start() None
class fireworks.core.tests.test_launchpad.LaunchPadOfflineTest(methodName='runTest')

Bases: TestCase

setUp() None

Hook method for setting up the test fixture before exercising it.

classmethod setUpClass() None

Hook method for setting up class fixture before running tests in the class.

tearDown() None

Hook method for deconstructing the test fixture after testing it.

classmethod tearDownClass() None

Hook method for deconstructing the class fixture after running all tests in the class.

test__recover_completed() None
test_recover_errors() None
class fireworks.core.tests.test_launchpad.LaunchPadRerunExceptionTest(methodName='runTest')

Bases: TestCase

setUp() None

Hook method for setting up the test fixture before exercising it.

classmethod setUpClass() None

Hook method for setting up class fixture before running tests in the class.

tearDown() None

Hook method for deconstructing the test fixture after testing it.

classmethod tearDownClass() None

Hook method for deconstructing the class fixture after running all tests in the class.

test_except_details_on_rerun() None
test_task_level_rerun() None
test_task_level_rerun_cp() None
test_task_level_rerun_prev_dir() None
class fireworks.core.tests.test_launchpad.LaunchPadTest(methodName='runTest')

Bases: TestCase

setUp() None

Hook method for setting up the test fixture before exercising it.

classmethod setUpClass() None

Hook method for setting up class fixture before running tests in the class.

tearDown() None

Hook method for deconstructing the test fixture after testing it.

classmethod tearDownClass() None

Hook method for deconstructing the class fixture after running all tests in the class.

test_add_wf() None
test_add_wfs() None
test_dict_from_file() None
test_pw_check() None
test_reset() None
class fireworks.core.tests.test_launchpad.WFLockTest(methodName='runTest')

Bases: TestCase

setUp() None

Hook method for setting up the test fixture before exercising it.

classmethod setUpClass() None

Hook method for setting up class fixture before running tests in the class.

tearDown() None

Hook method for deconstructing the test fixture after testing it.

classmethod tearDownClass() None

Hook method for deconstructing the class fixture after running all tests in the class.

test_fix_db_inconsistencies_completed() None
test_fix_db_inconsistencies_fizzled() None
class fireworks.core.tests.test_launchpad.WorkflowFireworkStatesTest(methodName='runTest')

Bases: TestCase

Class to test the firework states locally cached in workflow. The states have to be in sync with the actual firework state.

setUp() None

Hook method for setting up the test fixture before exercising it.

classmethod setUpClass() None

Hook method for setting up class fixture before running tests in the class.

tearDown() None

Hook method for deconstructing the test fixture after testing it.

classmethod tearDownClass() None

Hook method for deconstructing the class fixture after running all tests in the class.

test_archive_wf() None
test_defuse_fw() None
test_defuse_fw_after_completion() None
test_defuse_wf() None
test_reignite_fw() None
test_reignite_wf() None
test_rerun_fws() None
test_rerun_timed_fws() None

fireworks.core.tests.test_rocket module

class fireworks.core.tests.test_rocket.RocketTest(methodName='runTest')

Bases: TestCase

setUp() None

Hook method for setting up the test fixture before exercising it.

classmethod setUpClass() None

Hook method for setting up class fixture before running tests in the class.

tearDown() None

Hook method for deconstructing the test fixture after testing it.

classmethod tearDownClass() None

Hook method for deconstructing the class fixture after running all tests in the class.

test_postproc_exception() None
test_serializable_exception() None

fireworks.core.tests.test_tracker module

Tracker unitest.

class fireworks.core.tests.test_tracker.TrackerTest(methodName='runTest')

Bases: TestCase

setUp() None

Hook method for setting up the test fixture before exercising it.

classmethod setUpClass() None

Hook method for setting up class fixture before running tests in the class.

tearDown() None

Hook method for deconstructing the test fixture after testing it.

classmethod tearDownClass() None

Hook method for deconstructing the class fixture after running all tests in the class.

test_tracker() None

Launch a workflow and track the files.

test_tracker_failed_fw() None

Add a bad firetask to workflow and test the tracking.

test_tracker_mlaunch() None

Test the tracker for mlaunch.

Module contents