def test_worker_family_fit_function():
wf = WorkerFamily(**WF_DICT)
for child in wf.children:
child.run_simulation = lambda: None
child.get_simulation_error = lambda: 1
child.read_simulation_results = lambda: None
dummy_p_values = np.asarray([1])
assert (wf.fit(dummy_p_values) == len(
wf.children)), "Family fit function did not validate!"
def test_fig_initialization():
"""Tests the initialization of the optimization figure"""
# Monkey-patch the Worker's figure and plot initialization to make sure they're called.
def initialize_figure(self):
self.figure_method_called = True
def make_plots(self):
self.plot_method_called = True
old_initialize_figure = WorkerFamily.initialize_figure
WorkerFamily.initialize_figure = initialize_figure
old_make_plots = WorkerFamily.make_plots
WorkerFamily.make_plots = make_plots
WorkerFamily.figure_method_called = False
WorkerFamily.plot_method_called = False
this_wf_dict = copy.copy(WF_DICT)
this_wf_dict["display_progress"] = True
wf = WorkerFamily(**this_wf_dict)
assert (
wf.figure_method_called), "WorkerFamily figure method was not called!"
assert (
wf.plot_method_called), "WorkerFamily make plot method was not called!"
# Do some cleanup to fix the monkeypatching.
WorkerFamily.initialize_figure = old_initialize_figure
WorkerFamily.make_plots = old_make_plots
del WorkerFamily.figure_method_called
del WorkerFamily.plot_method_called
def test_worker_family_initializes_with_correct_target_data():
wf = WorkerFamily(**WF_DICT)
for i, child in enumerate(wf.children):
assert (child.target_data == i), (
"WorkerFamily did not initialize children with correct "
"target data!")
def test_worker_family_read_target_data_verifies_data_length_num_children_match_1(
):
this_wf_dict = copy.copy(WF_DICT)
this_wf_dict["target_data"] = np.ones(
(len(this_wf_dict["protocol_files"]), 3))
with pytest.raises(ValueError):
wf = WorkerFamily(**this_wf_dict)
def test_worker_family_initialization():
wf = WorkerFamily(**WF_DICT)
msg = "WorkerFamily did not initialize correctly!"
assert (wf.iteration_number == 0), msg
assert (wf.best_error == np.inf), msg
assert (wf.error_values == []), msg
assert (len(wf.children) == len(
WF_DICT["protocol_files"])), ("WorkerFamily did not initialize "
"correct number of children!")
def test_worker_family_read_target_data_array():
this_wf_dict = copy.copy(WF_DICT)
data = np.ones((len(WF_DICT["protocol_files"]), 2))
this_wf_dict["target_data"] = data
wf = WorkerFamily(**this_wf_dict)
assert (np.array_equal(
wf.target_data,
data)), "WorkerFamily did not read target_data correctly!"
def test_worker_family_read_target_data_string():
this_wf_dict = copy.copy(WF_DICT)
this_wf_dict["target_data"] = os.path.join(WF_DICT["output_dir"],
"wf_test_target_data.txt")
data = np.ones((len(WF_DICT["protocol_files"]), 2))
np.savetxt(this_wf_dict["target_data"], data)
wf = WorkerFamily(**this_wf_dict)
assert (np.array_equal(
wf.target_data,
data)), "WorkerFamily did not read target_data correctly!"
def test_worker_family_read_target_data_raises_type_error():
this_wf_dict = copy.copy(WF_DICT)
this_wf_dict["target_data"] = {"garbage": True}
with pytest.raises(TypeError):
wf = WorkerFamily(**this_wf_dict)