class TestGeneralMethods: def setUp(self): self.c = Component(["one", "two"]) self.c.one.free = False self.c.two.free = True self.c.one._number_of_elements = 1 self.c.two._number_of_elements = 2 def test_export_free(self): c = self.c c.one.export = mock.MagicMock() c.two.export = mock.MagicMock() c.free_parameters = {c.two, } call_args = {'folder': 'folder1', 'format': 'format1', 'save_std': 'save_std1'} c.export(only_free=True, **call_args) nt.assert_dict_equal(c.two.export.call_args[1], call_args) nt.assert_false(c.one.export.called) def test_export_all_no_twins(self): c = self.c c.one.export = mock.MagicMock() c.two.export = mock.MagicMock() c.free_parameters = {c.two, } call_args = {'folder': 'folder1', 'format': 'format1', 'save_std': 'save_std1'} c.export(only_free=False, **call_args) nt.assert_dict_equal(c.two.export.call_args[1], call_args) nt.assert_dict_equal(c.one.export.call_args[1], call_args) def test_export_all_twins(self): c = self.c c.one.export = mock.MagicMock() c.two.export = mock.MagicMock() c.two.twin = c.one c.free_parameters = {c.two, } call_args = {'folder': 'folder1', 'format': 'format1', 'save_std': 'save_std1'} c.export(only_free=False, **call_args) nt.assert_dict_equal(c.one.export.call_args[1], call_args) nt.assert_false(c.two.export.called) def test_update_number_parameters(self): self.c.nparam = 0 self.c.update_number_parameters() nt.assert_equal(self.c.nparam, 3) def test_fetch_from_array(self): arr = np.array([30, 20, 10]) arr_std = np.array([30.5, 20.5, 10.5]) self.c.fetch_values_from_array(arr, p_std=arr_std, onlyfree=False) nt.assert_equal(self.c.one.value, 30) nt.assert_equal(self.c.one.std, 30.5) nt.assert_equal(self.c.two.value, (20, 10)) nt.assert_equal(self.c.two.std, (20.5, 10.5)) def test_fetch_from_array_free(self): arr = np.array([30, 20, 10]) arr_std = np.array([30.5, 20.5, 10.5]) self.c.one.value = 1. self.c.one.std = np.nan self.c.fetch_values_from_array(arr, p_std=arr_std, onlyfree=True) nt.assert_equal(self.c.one.value, 1) nt.assert_is(self.c.one.std, np.nan) nt.assert_equal(self.c.two.value, (30, 20)) nt.assert_equal(self.c.two.std, (30.5, 20.5)) def test_fetch_stored_values_fixed(self): c = self.c c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values(only_fixed=True) nt.assert_true(c.one.fetch.called) nt.assert_false(c.two.fetch.called) def test_fetch_stored_values_all(self): c = self.c c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values() nt.assert_true(c.one.fetch.called) nt.assert_true(c.two.fetch.called) def test_fetch_stored_values_all_twinned_bad(self): c = self.c c.one._twin = 1. c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values() nt.assert_true(c.one.fetch.called) nt.assert_true(c.two.fetch.called) def test_fetch_stored_values_all_twinned(self): c = self.c c.one.twin = c.two c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values() nt.assert_false(c.one.fetch.called) nt.assert_true(c.two.fetch.called) def test_set_parameters_free_all(self): self.c.set_parameters_free() nt.assert_true(self.c.one.free) nt.assert_true(self.c.two.free) def test_set_parameters_free_name(self): self.c.set_parameters_free(['one']) nt.assert_true(self.c.one.free) nt.assert_true(self.c.two.free) def test_set_parameters_not_free_all(self): self.c.set_parameters_not_free() nt.assert_false(self.c.one.free) nt.assert_false(self.c.two.free) def test_set_parameters_not_free_name(self): self.c.one.free = True self.c.set_parameters_not_free(['two']) nt.assert_true(self.c.one.free) nt.assert_false(self.c.two.free)
class TestGeneralMethods: def setUp(self): self.c = Component(["one", "two"]) self.c.one.free = False self.c.two.free = True self.c.one._number_of_elements = 1 self.c.two._number_of_elements = 2 def test_export_free(self): c = self.c c.one.export = mock.MagicMock() c.two.export = mock.MagicMock() c.free_parameters = { c.two, } call_args = { 'folder': 'folder1', 'format': 'format1', 'save_std': 'save_std1' } c.export(only_free=True, **call_args) nt.assert_dict_equal(c.two.export.call_args[1], call_args) nt.assert_false(c.one.export.called) def test_export_all_no_twins(self): c = self.c c.one.export = mock.MagicMock() c.two.export = mock.MagicMock() c.free_parameters = { c.two, } call_args = { 'folder': 'folder1', 'format': 'format1', 'save_std': 'save_std1' } c.export(only_free=False, **call_args) nt.assert_dict_equal(c.two.export.call_args[1], call_args) nt.assert_dict_equal(c.one.export.call_args[1], call_args) def test_export_all_twins(self): c = self.c c.one.export = mock.MagicMock() c.two.export = mock.MagicMock() c.two.twin = c.one c.free_parameters = { c.two, } call_args = { 'folder': 'folder1', 'format': 'format1', 'save_std': 'save_std1' } c.export(only_free=False, **call_args) nt.assert_dict_equal(c.one.export.call_args[1], call_args) nt.assert_false(c.two.export.called) def test_update_number_parameters(self): self.c.nparam = 0 self.c.update_number_parameters() nt.assert_equal(self.c.nparam, 3) def test_fetch_from_array(self): arr = np.array([30, 20, 10]) arr_std = np.array([30.5, 20.5, 10.5]) self.c.fetch_values_from_array(arr, p_std=arr_std, onlyfree=False) nt.assert_equal(self.c.one.value, 30) nt.assert_equal(self.c.one.std, 30.5) nt.assert_equal(self.c.two.value, (20, 10)) nt.assert_equal(self.c.two.std, (20.5, 10.5)) def test_fetch_from_array_free(self): arr = np.array([30, 20, 10]) arr_std = np.array([30.5, 20.5, 10.5]) self.c.one.value = 1. self.c.one.std = np.nan self.c.fetch_values_from_array(arr, p_std=arr_std, onlyfree=True) nt.assert_equal(self.c.one.value, 1) nt.assert_is(self.c.one.std, np.nan) nt.assert_equal(self.c.two.value, (30, 20)) nt.assert_equal(self.c.two.std, (30.5, 20.5)) def test_fetch_stored_values_fixed(self): c = self.c c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values(only_fixed=True) nt.assert_true(c.one.fetch.called) nt.assert_false(c.two.fetch.called) def test_fetch_stored_values_all(self): c = self.c c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values() nt.assert_true(c.one.fetch.called) nt.assert_true(c.two.fetch.called) def test_fetch_stored_values_all_twinned_bad(self): c = self.c c.one._twin = 1. c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values() nt.assert_true(c.one.fetch.called) nt.assert_true(c.two.fetch.called) def test_fetch_stored_values_all_twinned(self): c = self.c c.one.twin = c.two c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values() nt.assert_false(c.one.fetch.called) nt.assert_true(c.two.fetch.called) def test_set_parameters_free_all(self): self.c.set_parameters_free() nt.assert_true(self.c.one.free) nt.assert_true(self.c.two.free) def test_set_parameters_free_name(self): self.c.set_parameters_free(['one']) nt.assert_true(self.c.one.free) nt.assert_true(self.c.two.free) def test_set_parameters_not_free_all(self): self.c.set_parameters_not_free() nt.assert_false(self.c.one.free) nt.assert_false(self.c.two.free) def test_set_parameters_not_free_name(self): self.c.one.free = True self.c.set_parameters_not_free(['two']) nt.assert_true(self.c.one.free) nt.assert_false(self.c.two.free)
class TestGeneralMethods: def setup_method(self, method): self.c = Component(["one", "two"]) self.c.one.free = False self.c.two.free = True self.c.one._number_of_elements = 1 self.c.two._number_of_elements = 2 def test_export_free(self): c = self.c c.one.export = mock.MagicMock() c.two.export = mock.MagicMock() c.free_parameters = { c.two, } call_args = { 'folder': 'folder1', 'format': 'format1', 'save_std': 'save_std1' } c.export(only_free=True, **call_args) assert c.two.export.call_args[1] == call_args assert not c.one.export.called def test_export_all_no_twins(self): c = self.c c.one.export = mock.MagicMock() c.two.export = mock.MagicMock() c.free_parameters = { c.two, } call_args = { 'folder': 'folder1', 'format': 'format1', 'save_std': 'save_std1' } c.export(only_free=False, **call_args) assert c.two.export.call_args[1] == call_args assert c.one.export.call_args[1] == call_args def test_export_all_twins(self): c = self.c c.one.export = mock.MagicMock() c.two.export = mock.MagicMock() c.two.twin = c.one c.free_parameters = { c.two, } call_args = { 'folder': 'folder1', 'format': 'format1', 'save_std': 'save_std1' } c.export(only_free=False, **call_args) assert c.one.export.call_args[1] == call_args assert not c.two.export.called def test_update_number_parameters(self): self.c.nparam = 0 self.c.update_number_parameters() assert self.c.nparam == 3 def test_fetch_from_array(self): arr = np.array([30, 20, 10]) arr_std = np.array([30.5, 20.5, 10.5]) self.c.fetch_values_from_array(arr, p_std=arr_std, onlyfree=False) assert self.c.one.value == 30 assert self.c.one.std == 30.5 assert self.c.two.value == (20, 10) assert self.c.two.std == (20.5, 10.5) def test_fetch_from_array_free(self): arr = np.array([30, 20, 10]) arr_std = np.array([30.5, 20.5, 10.5]) self.c.one.value = 1. self.c.one.std = np.nan self.c.fetch_values_from_array(arr, p_std=arr_std, onlyfree=True) assert self.c.one.value == 1 assert self.c.one.std is np.nan assert self.c.two.value == (30, 20) assert self.c.two.std == (30.5, 20.5) def test_fetch_stored_values_fixed(self): c = self.c c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values(only_fixed=True) assert c.one.fetch.called assert not c.two.fetch.called def test_fetch_stored_values_all(self): c = self.c c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values() assert c.one.fetch.called assert c.two.fetch.called def test_fetch_stored_values_all_twinned_bad(self): c = self.c c.one._twin = 1. c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values() assert c.one.fetch.called assert c.two.fetch.called def test_fetch_stored_values_all_twinned(self): c = self.c c.one.twin = c.two c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values() assert not c.one.fetch.called assert c.two.fetch.called def test_set_parameters_free_all(self): self.c.set_parameters_free() assert self.c.one.free assert self.c.two.free def test_set_parameters_free_name(self): self.c.set_parameters_free(['one']) assert self.c.one.free assert self.c.two.free def test_set_parameters_not_free_all(self): self.c.set_parameters_not_free() assert not self.c.one.free assert not self.c.two.free def test_set_parameters_not_free_name(self): self.c.one.free = True self.c.set_parameters_not_free(['two']) assert self.c.one.free assert not self.c.two.free
class TestGeneralMethods: def setup_method(self, method): self.c = Component(["one", "two"]) self.c.one.free = False self.c.two.free = True self.c.one._number_of_elements = 1 self.c.two._number_of_elements = 2 def test_export_free(self): c = self.c c.one.export = mock.MagicMock() c.two.export = mock.MagicMock() c.free_parameters = {c.two, } call_args = {'folder': 'folder1', 'format': 'format1', 'save_std': 'save_std1'} c.export(only_free=True, **call_args) assert c.two.export.call_args[1] == call_args assert not c.one.export.called def test_export_all_no_twins(self): c = self.c c.one.export = mock.MagicMock() c.two.export = mock.MagicMock() c.free_parameters = {c.two, } call_args = {'folder': 'folder1', 'format': 'format1', 'save_std': 'save_std1'} c.export(only_free=False, **call_args) assert c.two.export.call_args[1] == call_args assert c.one.export.call_args[1] == call_args def test_export_all_twins(self): c = self.c c.one.export = mock.MagicMock() c.two.export = mock.MagicMock() c.two.twin = c.one c.free_parameters = {c.two, } call_args = {'folder': 'folder1', 'format': 'format1', 'save_std': 'save_std1'} c.export(only_free=False, **call_args) assert c.one.export.call_args[1] == call_args assert not c.two.export.called def test_update_number_parameters(self): self.c.nparam = 0 self.c.update_number_parameters() assert self.c.nparam == 3 def test_fetch_from_array(self): arr = np.array([30, 20, 10]) arr_std = np.array([30.5, 20.5, 10.5]) self.c.fetch_values_from_array(arr, p_std=arr_std, onlyfree=False) assert self.c.one.value == 30 assert self.c.one.std == 30.5 assert self.c.two.value == (20, 10) assert self.c.two.std == (20.5, 10.5) def test_fetch_from_array_free(self): arr = np.array([30, 20, 10]) arr_std = np.array([30.5, 20.5, 10.5]) self.c.one.value = 1. self.c.one.std = np.nan self.c.fetch_values_from_array(arr, p_std=arr_std, onlyfree=True) assert self.c.one.value == 1 assert self.c.one.std is np.nan assert self.c.two.value == (30, 20) assert self.c.two.std == (30.5, 20.5) def test_fetch_stored_values_fixed(self): c = self.c c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values(only_fixed=True) assert c.one.fetch.called assert not c.two.fetch.called def test_fetch_stored_values_all(self): c = self.c c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values() assert c.one.fetch.called assert c.two.fetch.called def test_fetch_stored_values_all_twinned_bad(self): c = self.c c.one._twin = 1. c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values() assert c.one.fetch.called assert c.two.fetch.called def test_fetch_stored_values_all_twinned(self): c = self.c c.one.twin = c.two c.one.fetch = mock.MagicMock() c.two.fetch = mock.MagicMock() c.fetch_stored_values() assert not c.one.fetch.called assert c.two.fetch.called def test_set_parameters_free_all(self): self.c.set_parameters_free() assert self.c.one.free assert self.c.two.free def test_set_parameters_free_name(self): self.c.set_parameters_free(['one']) assert self.c.one.free assert self.c.two.free def test_set_parameters_not_free_all(self): self.c.set_parameters_not_free() assert not self.c.one.free assert not self.c.two.free def test_set_parameters_not_free_name(self): self.c.one.free = True self.c.set_parameters_not_free(['two']) assert self.c.one.free assert not self.c.two.free