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
