def test_not_adding_pars_during_single_run(self):
traj = Trajectory()
traj._is_run = True
with self.assertRaises(TypeError):
traj.f_add_parameter('dd')
with self.assertRaises(TypeError):
traj.f_add_parameter_group('dd')
with self.assertRaises(TypeError):
traj.f_add_config('dd')
with self.assertRaises(TypeError):
traj.f_add_config_group('dd')
class TrajectoryTest(unittest.TestCase):
def setUp(self):
name = 'Moop'
self.traj = Trajectory(name,dynamically_imported_classes=[ImAParameterInDisguise,
'pypet.tests.test_helpers.ImAResultInDisguise'])
self.assertTrue(self.traj.f_is_empty())
comment = 'This is a comment'
self.traj.v_comment=comment
self.assertTrue(comment == self.traj.v_comment)
self.traj.f_add_parameter('IntParam',3)
sparsemat = spsp.csr_matrix((1000,1000))
sparsemat[1,2] = 17.777
self.traj.f_add_parameter(PickleParameter,'SparseParam', sparsemat)
self.traj.f_add_parameter('FloatParam')
self.traj.f_add_derived_parameter(Parameter('FortyTwo', 42))
self.traj.f_add_result(Result,'Im.A.Simple.Result',44444)
self.traj.FloatParam=4.0
self.explore_dict = {'FloatParam':[1.0,1.1,1.2,1.3]}
self.traj.f_explore(self.explore_dict)
self.assertTrue(len(self.traj) == 4)
self.traj.f_add_parameter_group('peter.paul')
self.traj.f_add_parameter('peter.markus.yve',6)
self.traj.f_add_result('Test',42)
self.traj.peter.f_add_parameter('paul.peter')
self.traj.f_add_config('make.impossible.promises',1)
with self.assertRaises(AttributeError):
self.traj.markus.peter
with self.assertRaises(ValueError):
self.traj.f_add_parameter('Peter. h ._hurz')
def test_shrink(self):
self.assertTrue(len(self.traj)>1)
self.traj.f_shrink()
self.assertTrue(len(self.traj)==1)
self.assertTrue(len(self.traj.f_get_explored_parameters())==0)
def test_get_all(self):
all_nodes = self.traj.f_get_all('peter')
self.assertTrue(len(all_nodes)==2)
all_nodes = self.traj.f_get_all('peter.yve')
self.assertTrue(len(all_nodes)==1)
all_nodes = self.traj.f_get_all('paul.yve')
self.assertTrue(len(all_nodes)==0)
self.traj.f_add_parameter('paul.paul')
all_nodes = self.traj.peter.f_get_all('paul')
self.assertTrue(len(all_nodes)==1)
self.traj.f_add_result('results.runs.run_00000000.x.y',10)
self.traj.f_add_result('results.runs.run_00000001.x.y',10)
self.traj.f_add_derived_parameter('hfhfhf.x.y')
self.traj.f_add_result('x.y.y')
self.traj.f_as_run(1)
all_nodes=self.traj.f_get_all('x.y')
self.assertTrue(len(all_nodes)==4)
def test_backwards_search(self):
x=self.traj.peter.f_get('paul.peter', backwards_search=False)
y=self.traj.f_get('peter.peter', backwards_search=True)
self.assertTrue(x is y)
#
# def test_value_error_on_search_strategy_assignment(self):
# with self.assertRaises(ValueError):
# self.traj.v_search_strategy = 'ewnforenfre'
def test_get_data_dictionaries_directly(self):
############## Cofig ###################
config_dict_from_subtree = self.traj.config.f_to_dict()
self.assertTrue(len(config_dict_from_subtree)>0)
config_dict_directly = self.traj.f_get_config(copy=True)
self.assertTrue(comp.nested_equal(config_dict_directly,config_dict_from_subtree),
'%s!=%s' % (str(config_dict_directly),str(config_dict_directly)))
config_dict_directly = self.traj.f_get_config(copy=False)
self.assertTrue(comp.nested_equal(config_dict_directly,config_dict_from_subtree),
'%s!=%s' % (str(config_dict_directly),str(config_dict_directly)))
config_dict_from_subtree = self.traj.config.f_to_dict(fast_access=True)
with self.assertRaises(ValueError):
config_dict_directly = self.traj.f_get_config(copy=False, fast_access=True)
config_dict_directly = self.traj.f_get_config(copy=True, fast_access=True)
self.assertTrue(comp.nested_equal(config_dict_directly,config_dict_from_subtree),
'%s!=%s' % (str(config_dict_directly),str(config_dict_directly)))
############## Parameters #############################
parameters_dict_from_subtree = self.traj.parameters.f_to_dict()
self.assertTrue(len(parameters_dict_from_subtree)>0)
parameters_dict_directly = self.traj.f_get_parameters(copy=True)
self.assertTrue(comp.nested_equal(parameters_dict_directly,parameters_dict_from_subtree),
'%s!=%s' % (str(parameters_dict_directly),str(parameters_dict_directly)))
parameters_dict_directly = self.traj.f_get_parameters(copy=False)
self.assertTrue(comp.nested_equal(parameters_dict_directly,parameters_dict_from_subtree),
'%s!=%s' % (str(parameters_dict_directly),str(parameters_dict_directly)))
### Empty Parameters won't support fast access so we need to set
self.traj.paul.peter.f_set(42)
parameters_dict_from_subtree = self.traj.parameters.f_to_dict(fast_access=True)
with self.assertRaises(ValueError):
parameters_dict_directly = self.traj.f_get_parameters(copy=False, fast_access=True)
parameters_dict_directly = self.traj.f_get_parameters(copy=True, fast_access=True)
self.assertTrue(comp.nested_equal(parameters_dict_directly,parameters_dict_from_subtree),
'%s!=%s' % (str(parameters_dict_directly),str(parameters_dict_directly)))
################# Derived Parameters ############################
derived_parameters_dict_from_subtree = self.traj.derived_parameters.f_to_dict()
self.assertTrue(len(derived_parameters_dict_from_subtree)>0)
derived_parameters_dict_directly = self.traj.f_get_derived_parameters(copy=True)
self.assertTrue(comp.nested_equal(derived_parameters_dict_directly,derived_parameters_dict_from_subtree),
'%s!=%s' % (str(derived_parameters_dict_directly),str(derived_parameters_dict_directly)))
derived_parameters_dict_directly = self.traj.f_get_derived_parameters(copy=False)
self.assertTrue(comp.nested_equal(derived_parameters_dict_directly,derived_parameters_dict_from_subtree),
'%s!=%s' % (str(derived_parameters_dict_directly),str(derived_parameters_dict_directly)))
derived_parameters_dict_from_subtree = self.traj.derived_parameters.f_to_dict(fast_access=True)
with self.assertRaises(ValueError):
derived_parameters_dict_directly = self.traj.f_get_derived_parameters(copy=False, fast_access=True)
derived_parameters_dict_directly = self.traj.f_get_derived_parameters(copy=True, fast_access=True)
self.assertTrue(comp.nested_equal(derived_parameters_dict_directly,derived_parameters_dict_from_subtree),
'%s!=%s' % (str(derived_parameters_dict_directly),str(derived_parameters_dict_directly)))
############## Results #################################
results_dict_from_subtree = self.traj.results.f_to_dict()
self.assertTrue(len(results_dict_from_subtree)>0)
results_dict_directly = self.traj.f_get_results(copy=True)
self.assertTrue(comp.nested_equal(results_dict_directly,results_dict_from_subtree),
'%s!=%s' % (str(results_dict_directly),str(results_dict_directly)))
results_dict_directly = self.traj.f_get_results(copy=False)
self.assertTrue(comp.nested_equal(results_dict_directly,results_dict_from_subtree),
'%s!=%s' % (str(results_dict_directly),str(results_dict_directly)))
results_dict_from_subtree = self.traj.results.f_to_dict(fast_access=True)
with self.assertRaises(ValueError):
results_dict_directly = self.traj.f_get_results(copy=False, fast_access=True)
results_dict_directly = self.traj.f_get_results(copy=True, fast_access=True)
self.assertTrue(comp.nested_equal(results_dict_directly,results_dict_from_subtree),
'%s!=%s' % (str(results_dict_directly),str(results_dict_directly)))
################### Explored Parameters #######################
# #We need to unlock the parameter because we have accessed it above
# self.traj.f_get('yve').f_unlock()
# explore_dict = {'yve':[4,5,65,66]}
# # We can add to existing explored parameters if we match the length
# self.traj.f_expand(explore_dict)
explore_dict_directly = self.traj.f_get_explored_parameters(copy=False)
for key in self.explore_dict:
self.assertTrue(comp.nested_equal(self.traj.f_get(key),
explore_dict_directly[self.traj.f_get(key).v_full_name]))
explore_dict_directly = self.traj.f_get_explored_parameters(copy=True)
for key in self.explore_dict:
self.assertTrue(comp.nested_equal(self.traj.f_get(key),
explore_dict_directly[self.traj.f_get(key).v_full_name]))
with self.assertRaises(ValueError):
explore_dict_directly = self.traj.f_get_explored_parameters(copy=False, fast_access=True)
explore_dict_directly = self.traj.f_get_explored_parameters(copy=True, fast_access=True)
for key in self.explore_dict:
self.assertTrue(comp.nested_equal(self.traj.f_get(key,fast_access=True),
explore_dict_directly[self.traj.f_get(key).v_full_name]))
def test_increase_exploration(self):
self.explore_dict = {'IntParam':[2,1,1,3]}
self.traj.f_explore(self.explore_dict)
self.assertTrue(len(self.traj._explored_parameters)==2)
self.traj._stored=True
with self.assertRaises(TypeError):
self.traj.f_explore(self.explore_dict)
def test_f_get(self):
self.traj.v_fast_access=True
self.traj.f_get('FloatParam', fast_access=True) == self.traj.FloatParam
self.traj.v_fast_access=False
self.assertTrue(self.traj.f_get('FloatParam').f_get() == 4.0 , '%d != 4.0' %self.traj.f_get('FloatParam').f_get())
self.assertTrue(self.traj.FortyTwo.f_get() == 42)
def test_get_item(self):
self.assertEqual(self.traj.markus.yve, self.traj['markus.yve'])
self.assertEqual(self.traj.markus.yve, self.traj['markus','yve'])
# def test_round_brackets(self):
# x='markus'
#
# y='yve'
#
# self.assertEqual(self.traj(x)(y), self.traj.markus.f_get('yve'))
@staticmethod
def get_depth_dict(traj, as_run=None):
bfs_queue=[traj]
depth_dict={}
while len(bfs_queue)>0:
item = bfs_queue.pop(0)
depth = item.v_depth
if not depth in depth_dict:
depth_dict[depth]=[]
depth_dict[depth].append(item)
if not item.v_is_leaf:
if as_run in item._children:
bfs_queue.append(item._children[as_run])
for child in item._children.values():
if not (child.v_name.startswith(pypetconstants.RUN_NAME) and
child.v_name != pypetconstants.RUN_NAME_DUMMY):
bfs_queue.append(child)
else:
for child in item._children.values():
bfs_queue.append(child)
return depth_dict
def test_iter_bfs(self):
depth_dict = self.get_depth_dict(self.traj)
depth_dict[0] =[]
prev_depth = 0
for node in self.traj.f_iter_nodes(recursive=True):
if prev_depth != node.v_depth:
self.assertEqual(len(depth_dict[prev_depth]),0)
prev_depth = node.v_depth
depth_dict[node.v_depth].remove(node)
depth_dict = self.get_depth_dict(self.traj)
depth_dict[0] =[]
prev_depth = 0
self.traj.v_iter_recursive = True
for node in self.traj:
if prev_depth != node.v_depth:
self.assertEqual(len(depth_dict[prev_depth]),0)
prev_depth = node.v_depth
depth_dict[node.v_depth].remove(node)
#
#
# def test_iter_dfs(self):
#
# prev_node = None
#
# x= [x for x in self.traj.f_iter_nodes(recursive=True, search_strategy='DFS')]
#
# for node in self.traj.f_iter_nodes(recursive=True, search_strategy='DFS'):
# if not prev_node is None:
# if not prev_node.v_is_leaf and len(prev_node._children) > 0:
# self.assertTrue(node.v_name in prev_node._children)
#
# prev_node = node
#
# prev_node = None
#
# self.traj.v_iter_recursive=True
# self.traj.v_search_strategy='DFS'
#
# y = [y for y in self.traj]
#
# self.assertEqual(x,y)
#
# for node in self.traj:
# if not prev_node is None:
# if not prev_node.v_is_leaf and len(prev_node._children) > 0:
# self.assertTrue(node.v_name in prev_node._children)
#
# prev_node = node
def test_iter_bfs_as_run(self):
as_run = 1
self.traj.f_add_result('results.run_00000000.resulttest', 42)
self.traj.f_add_result('results.run_00000001.resulttest', 43)
self.traj.f_as_run(as_run)
depth_dict = self.get_depth_dict(self.traj, self.traj.f_idx_to_run(as_run))
depth_dict[0] =[]
prev_depth = 0
for node in self.traj.f_iter_nodes(recursive=True):
self.assertTrue('run_00000000' not in node.v_full_name)
if prev_depth != node.v_depth:
self.assertEqual(len(depth_dict[prev_depth]),0)
prev_depth = node.v_depth
depth_dict[node.v_depth].remove(node)
# def test_iter_dfs_as_run(self):
#
# self.traj.f_add_result('results.run_00000000.resulttest', 42)
# self.traj.f_add_result('results.run_00000001.resulttest', 43)
#
# self.traj.f_as_run('run_00000001')
#
# prev_node = None
#
# x= [x for x in self.traj.f_iter_nodes(recursive=True, search_strategy='DFS')]
#
# for node in self.traj.f_iter_nodes(recursive=True, search_strategy='DFS'):
# self.assertTrue('run_00000000' not in node.v_full_name)
#
# if not prev_node is None:
# if not prev_node.v_is_leaf and len(prev_node._children) > 0:
# self.assertTrue(node.v_name in prev_node._children)
#
# prev_node = node
def test_find_in_all_runs(self):
self.traj.f_add_result('results.runs.run_00000000.sub.resulttest', 42)
self.traj.f_add_result('results.runs.run_00000001.sub.resulttest', 43)
self.traj.f_add_result('results.runs.run_00000002.sub.resulttest', 44)
self.traj.f_add_result('results.runs.run_00000002.sub.resulttest2', 42)
self.traj.f_add_result('results.runs.run_00000003.sub.resulttest2', 43)
self.traj.f_add_derived_parameter('derived_parameters.runs.run_00000002.testing', 44)
res_dict = self.traj.f_get_from_runs('kkkkkkdjfoiuref')
self.assertTrue(len(res_dict)==0)
res_dict = self.traj.f_get_from_runs('resulttest', fast_access=True)
self.assertTrue(len(res_dict)==3)
self.assertTrue(res_dict['run_00000001']==43)
self.assertTrue('run_00000003' not in res_dict)
res_dict = self.traj.f_get_from_runs(name='sub.resulttest2', use_indices=True)
self.assertTrue(len(res_dict)==2)
self.assertTrue(res_dict[3]is self.traj.f_get('run_00000003.resulttest2'))
self.assertTrue(1 not in res_dict)
res_dict = self.traj.f_get_from_runs(name='testing', where='derived_parameters')
self.assertTrue(len(res_dict)==1)
self.traj.f_add_result('results.runs.run_00000002.sub.sub.sub.sub.resulttest2', 444)
self.traj.f_add_result('results.runs.run_00000002.sub.sub.sub.resulttest2', 444)
with self.assertRaises(pex.NotUniqueNodeError):
self.traj.f_get_from_runs('sub.sub.resulttest2', backwards_search=True)
with self.assertRaises(ValueError):
self.traj.f_get_from_runs('test', where='Portland')
def test_illegal_namings(self):
self.traj=Trajectory('resulttest2')
with self.assertRaises(ValueError):
self.traj.f_add_parameter('f_get')
with self.assertRaises(ValueError):
self.traj.f_add_result('test.$.k.$')
rg=self.traj.f_add_result_group('ggg.$')
with self.assertRaises(ValueError):
rg.f_add_result('$.fff')
self.traj.f_add_result_group('test.$.k')
with self.assertRaises(ValueError):
self.traj.res.k.f_add_result('$.jjj')
with self.assertRaises(ValueError):
self.traj.f_add_parameter('e'*129)
with self.assertRaises(ValueError):
self.traj.f_add_parameter('e'*120+'.j'*120+'.k'*40)
with self.assertRaises(ValueError):
self.traj.f_add_parameter('crun',22)
def test_max_depth(self):
self.traj.f_add_parameter('halo.this.is.a.depth.testrr')
contains = self.traj.f_contains('a.depth.testrr', max_depth=3)
self.assertFalse(contains)
contains = self.traj.par.f_contains('halo.this.is.a.depth.testrr', shortcuts=False)
self.assertTrue(contains)
contains = self.traj.par.f_contains('halo.this.depth.testrr', shortcuts=True)
self.assertTrue(contains)
contains = self.traj.par.f_contains('testrr', shortcuts=True)
self.assertTrue(contains)
contains = 'testrr' in self.traj
self.assertTrue(contains)
contains = self.traj.par.f_contains('testrr', max_depth=5)
self.assertFalse(contains)
self.traj.v_max_depth = 5
contains = 'testrr' in self.traj
self.assertFalse(contains)
def test_root_getting(self):
traj = Trajectory()
traj.f_add_config_group('ff')
root = traj.ff.f_get_root()
self.assertTrue(root is traj)
def test_not_adding_pars_during_single_run(self):
traj = Trajectory()
traj._is_run = True
with self.assertRaises(TypeError):
traj.f_add_parameter('dd')
with self.assertRaises(TypeError):
traj.f_add_parameter_group('dd')
with self.assertRaises(TypeError):
traj.f_add_config('dd')
with self.assertRaises(TypeError):
traj.f_add_config_group('dd')
def test_attribute_error_raises_when_leaf_and_group_with_same_name_are_added(self):
self.traj = Trajectory()
self.traj.f_add_parameter('test.param1')
with self.assertRaises(AttributeError):
self.traj.f_add_parameter('test.param1.param2')
with self.assertRaises(AttributeError):
self.traj.f_add_parameter('test')
def testremove(self):
self.traj.f_remove_item(self.traj.f_get('peter.markus.yve'),remove_empty_groups=True)
with self.assertRaises(AttributeError):
self.traj.peter.markus.yve
self.assertFalse('peter.markus.yve' in self.traj)
#self.assertTrue(len(self.traj)==1)
self.traj.f_remove_item('FortyTwo',remove_empty_groups=True)
self.traj.f_remove_item('SparseParam')
self.traj.f_remove_item('IntParam')
#self.assertTrue(len(self.traj)==1)
def test_changing(self):
self.traj.f_preset_config('testconf', 1)
self.traj.f_preset_parameter('testparam', 1)
self.traj.f_preset_parameter('I_do_not_exist', 2)
self.traj.f_add_parameter('testparam', 0)
self.traj.f_add_config('testconf', 0)
self.traj.v_fast_access=True
self.assertTrue(self.traj.testparam == 1)
self.assertTrue(self.traj.testconf == 1)
### should raise an error because 'I_do_not_exist', does not exist:
with self.assertRaises(pex.PresettingError):
self.traj._prepare_experiment()
def test_f_is_completed(self):
traj = Trajectory()
traj.f_add_parameter('test', 42)
traj.f_explore({'test':[1,2,3,4]})
self.assertFalse(traj.f_is_completed())
for run_name in traj.f_get_run_names():
self.assertFalse(traj.f_is_completed(run_name))
traj._run_information[traj.f_idx_to_run(1)]['completed']=1
self.assertFalse(traj.f_is_completed())
self.assertTrue(traj.f_is_completed(1))
for run_name in traj.f_get_run_names():
traj._run_information[run_name]['completed']=1
self.assertTrue(traj.f_is_completed())
for run_name in traj.f_get_run_names():
self.assertTrue(traj.f_is_completed(run_name))
def test_if_picklable(self):
self.traj.v_fast_access=True
#self.traj.v_full_copy=True
dump = pickle.dumps(self.traj)
newtraj = pickle.loads(dump)
self.assertTrue(len(newtraj) == len(self.traj))
new_items = newtraj.f_to_dict(fast_access=True)
for key, val in self.traj.f_to_dict(fast_access=True).items():
#val = newtraj.f_get(table_name)
nval = new_items[key]
if isinstance(val, BaseResult):
for ikey in val._data:
self.assertTrue(str(nval.f_get(ikey))==str(val.f_get(ikey)))
else:
self.assertTrue(str(val)==str(nval), '%s != %s' %(str(val),str(nval)))
def test_dynamic_class_loading(self):
self.traj.f_add_parameter(ImAParameterInDisguise,'Rolf', 1.8)
def test_standard_change_param_change(self):
self.traj.v_standard_parameter=ImAParameterInDisguise
self.traj.f_add_parameter('I.should_be_not.normal')
self.assertIsInstance(self.traj.f_get('normal'), ImAParameterInDisguise,'Param is %s insted of ParamInDisguise.' %str(type(self.traj.normal)))
self.traj.v_standard_result=ImAResultInDisguise
self.traj.f_add_result('Peter.Parker')
self.assertIsInstance(self.traj.Parker, ImAResultInDisguise)
def test_remove_of_explored_stuff_if_saved(self):
self.traj = Trajectory()
self.traj.f_add_parameter('test', 42)
self.traj.f_explore({'test':[1,2,3,4]})
self.traj._stored=True
self.traj.parameters.f_remove_child('test')
len(self.traj) == 4
def test_remov_of_explored_stuff_if_not_saved(self):
self.traj = Trajectory()
self.traj.f_add_parameter('test', 42)
self.traj.f_explore({'test':[1,2,3,4]})
self.traj.parameters.f_remove_child('test')
self.assertTrue(len(self.traj) == 1)
def test_not_unique_search(self):
self.traj = Trajectory()
self.traj.f_add_parameter('ghgghg.test')
self.traj.f_add_parameter('ghdsfdfdsfdsghg.test')
with self.assertRaises(pex.NotUniqueNodeError):
self.traj.test
self.traj.f_add_parameter('depth0.depth1.depth2.findme', 42)
self.traj.f_add_parameter('depth0.depth1.findme', 43)
self.assertTrue(self.traj.findme==43)
with self.assertRaises(pex.NotUniqueNodeError):
self.traj.f_get('depth0.findme', backwards_search=True)
def test_contains_item_identity(self):
peterpaul = self.traj.f_get('peter.paul')
self.assertTrue(peterpaul in self.traj)
peterpaulcopy = copy.deepcopy(peterpaul)
self.assertFalse(peterpaulcopy in self.traj)
def test_get_children(self):
for node in self.traj.f_iter_nodes():
if not node.v_is_leaf:
self.assertEqual(id(node.f_get_children(copy=False)), id(node._children))
self.assertNotEqual(id(node.f_get_children(copy=True)), id(node._children))
self.assertEqual(sorted(node.f_get_children(copy=True).keys()),
sorted(node._children.keys()))
def test_short_cuts(self):
self.traj = Trajectory()
self.traj.f_add_parameter('test', 42)
self.traj.f_add_config('tefffst', 42)
self.traj.f_add_derived_parameter('dtest', 42)
self.traj.f_add_result('safd', 42)
self.traj.f_explore({'test':[1,2,3,4]})
self.assertEqual(id(self.traj.par), id(self.traj.parameters))
#self.assertEqual(id(self.traj.param), id(self.traj.parameters))
self.assertEqual(id(self.traj.dpar), id(self.traj.derived_parameters))
#self.assertEqual(id(self.traj.dparam), id(self.traj.derived_parameters))
self.assertEqual(id(self.traj.conf), id(self.traj.config))
self.assertEqual(id(self.traj.res), id(self.traj.results))
srun = self.traj._make_single_run(3)
srun.f_add_result('sdffds',42)
self.assertEqual(id(srun.results.crun), id(srun.results.f_get(srun.v_name)))