def test_delete_links(self): traj = Trajectory(name='TestDelete', filename=make_temp_dir('testpartiallydel.hdf5'), add_time=True) res = traj.f_add_result('mytest.test', a='b', c='d') traj.f_add_link('x.y', res) traj.f_add_link('x.g.h', res) traj.f_store() traj.f_remove_child('x', recursive=True) traj.f_load() self.assertEqual(traj.x.y.a, traj.test.a) self.assertEqual(traj.x.g.h.a, traj.test.a) traj.f_delete_link('x.y', remove_from_trajectory=True) traj.f_delete_link((traj.x.g, 'h'), remove_from_trajectory=True) traj.f_load() with self.assertRaises(AttributeError): traj.x.g.h
def test_storage_and_loading(self): filename = make_temp_dir('linktest.hdf5') traj = Trajectory(filename=filename) traj.f_add_parameter_group('test') traj.f_add_parameter_group('test2') res = traj.f_add_result('kk', 42) traj.par.f_add_link('gg', res) traj.f_add_link('hh', res) traj.f_add_link('jj', traj.par) traj.f_add_link('ii', res) traj.test.f_add_link('circle1', traj.test2) traj.test2.f_add_link('circle2', traj.test) traj.test.f_add_link('circle2', traj.test.circle1.circle2) traj.f_add_parameter_group('test.ab.bc.cd') traj.cd.f_add_link(traj.test) traj.test.f_add_link(traj.cd) traj.f_store() traj2 = Trajectory(filename=filename) traj2.f_load(name=traj.v_name, load_data=2) self.assertTrue(traj.kk == traj2.gg, '%s != %s' % (traj.kk, traj2.gg)) self.assertTrue(traj.cd.test is traj.test) self.assertTrue(len(traj._linked_by), len(traj2._linked_by)) self.compare_trajectories(traj, traj2) self.assertTrue('jj' in traj2._nn_interface._links_count) traj2.f_remove_child('jj') self.assertTrue('jj' not in traj2._nn_interface._links_count) traj2.f_remove_child('hh') traj2.f_remove_child('ii') traj2.f_remove_child('parameters', recursive=True) traj2.v_auto_load = True group = traj2.par.test2.circle2 self.assertTrue(group is traj2.test) retest = traj2.test.circle1 self.assertTrue(retest is traj2.test2) self.assertTrue(traj2.test.circle2 is traj2.test) self.assertTrue(traj2.hh == traj2.res.kk) traj2.v_auto_load = False traj2.f_load_child('jj') self.assertTrue(traj2.jj is traj2.par) traj2.f_load(load_data=2) self.assertTrue(traj2.ii == traj2.res.kk)
def test_link_creation(self): traj = Trajectory() traj.f_add_parameter_group('test.test3') traj.f_add_parameter_group('test2') with self.assertRaises(AttributeError): traj.par.f_add_link('test', traj.test2) with self.assertRaises(ValueError): traj.f_add_link('parameters', traj.test) with self.assertRaises(ValueError): traj.f_add_link('kkkk', PickleResult('fff', 555)) traj.test.f_add_link('circle1', traj.test2) traj.test2.f_add_link('circle2', traj.test) self.assertTrue(traj.test.circle1.circle2.circle1.circle2 is traj.test) traj.f_add_link('hh', traj.test) traj.par.f_add_link('overview', traj.test) with self.assertRaises(ValueError): traj.f_add_link('overview', traj.test) with self.assertRaises(ValueError): traj.f_add_link('v_crun') with self.assertRaises(ValueError): traj.par.f_add_link('gg', traj) with self.assertRaises(AttributeError): traj.f_add_parameter('test.circle1.testy', 33) traj.par.f_add_link('gg', traj.circle1) self.assertTrue(traj.gg is traj.test2) self.assertTrue(traj.test2.test3 is traj.par.test.test3) traj.f_add_link(traj.test3) self.assertTrue('test3' in traj._links)
def test_new_assignment_method(self): filename = make_temp_dir('newassignment.hdf5') traj = Trajectory(filename=filename) traj.v_lazy_adding = True comment = 'A number' traj.par.x = 44, comment self.assertTrue(traj.f_get('x').v_comment == comment) traj.par.iamgroup = a_new_group self.assertTrue(isinstance(traj.iamgroup, ParameterGroup)) traj.v_lazy_adding = False traj.x = 45 self.assertTrue(traj.par.f_get('x').f_get() == 45) self.assertTrue(isinstance(traj.f_get('x'), Parameter)) traj.f = Parameter('lll', 444, 'lll') self.assertTrue(traj.f_get('f').v_name == 'f') traj.v_lazy_adding = True traj.res.k = 22, 'Hi' self.assertTrue(isinstance(traj.f_get('k'), Result)) self.assertTrue(traj.f_get('k')[1] == 'Hi') with self.assertRaises(AttributeError): traj.res.k = 33, 'adsd' conf = traj.conf with self.assertRaises(AttributeError): conf = traj.conf.jjjj traj.f_set_properties(fast_access=True) traj.crun = 43, 'JJJ' self.assertTrue(traj.run_A[0] == 43) with self.assertRaises(AttributeError): traj.f_set_properties(j=7) with self.assertRaises(AttributeError): traj.f_set_properties(depth=7) traj.hui = (('444', 'kkkk',), 'l') self.assertTrue(traj.f_get('hui')[1] == 'l') with self.assertRaises(AttributeError): traj.hui = ('445', 'kkkk',) traj.f_get('hui').f_set(('445', 'kkkk',)) self.assertTrue(traj.f_get('hui')[1] == 'l') self.assertTrue(traj.hui[0] == ('445', 'kkkk',)) traj.f_add_link('klkikju', traj.par) # for shizzle traj.meee = Result('h', 43, hui = 3213, comment='du') self.assertTrue(traj.meee.h.h == 43) with self.assertRaises(TypeError): traj.par.mu = NNGroupNode('jj', comment='mi') with self.assertRaises(TypeError): traj.res.mu = NNGroupNode('jj', comment='mi') with self.assertRaises(TypeError): traj.conf.mu = NNGroupNode('jj', comment='mi') with self.assertRaises(TypeError): traj.dpar.mu = NNGroupNode('jj', comment='mi') with self.assertRaises(TypeError): traj.par.mu = ResultGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.dpar.mu = ResultGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.conf.mu = ResultGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.mu = ResultGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.par.mu = ConfigGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.dpar.mu = ConfigGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.res.mu = ConfigGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.mu = ConfigGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.par.mu = DerivedParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.conf.mu = DerivedParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.res.mu = DerivedParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.mu = DerivedParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.dpar.mu = ParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.res.mu = ParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.conf.mu = ParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.mu = ParameterGroup('jj', comment='mi') traj.par.mu = ParameterGroup('jj', comment='mi') traj.res.mus = ResultGroup('jj', comment='mi') traj.mu = NNGroupNode('jj') cg = ConfigGroup('a.g') traj.conf.a = cg self.assertTrue(traj.f_get('conf.a.a.g', shortcuts=False) is cg) dg = DerivedParameterGroup('ttt') traj.dpar.ttt = dg self.assertTrue(traj.f_get('dpar.ttt', shortcuts=False) is dg) traj.mylink = traj.par self.assertTrue(traj.mylink is traj.par) traj.vvv = NNGroupNode('', comment='kkk') self.assertTrue(traj.vvv.v_full_name == 'vvv') self.assertTrue(traj.par.mu.v_name == 'mu') traj.rrr = MyParamGroup('ff') traj.par.g = MyParamGroup('') pg = traj.f_add_parameter_group(comment='gg', full_name='me') self.assertTrue(traj.par.me is pg) traj.f_store() traj = load_trajectory(index=-1, filename=filename, dynamic_imports=MyParamGroup) self.assertTrue(isinstance(traj.rrr, NNGroupNode)) self.assertTrue(isinstance(traj.rrr.ff, MyParamGroup)) self.assertTrue(isinstance(traj.par.g, MyParamGroup)) traj.par = Parameter('hiho', 42, comment='you') traj.par = Parameter('g1.g2.g3.g4.g5', 43) self.assertTrue(traj.hiho == 42) self.assertTrue(isinstance(traj.par.g1, ParameterGroup )) self.assertTrue(isinstance(traj.par.g3, ParameterGroup )) self.assertTrue(traj.g3.g5 == 43)
def test_link_creation(self): traj = Trajectory() traj.f_add_parameter_group('test.test3') traj.f_add_parameter_group('test2') with self.assertRaises(AttributeError): traj.par.f_add_link('test', traj.test2) with self.assertRaises(ValueError): traj.f_add_link('parameters', traj.test) with self.assertRaises(ValueError): traj.f_add_link('kkkk', PickleResult('fff', 555)) traj.test.f_add_link('circle1' , traj.test2) traj.test2.f_add_link('circle2' , traj.test) self.assertTrue(traj.test.circle1.circle2.circle1.circle2 is traj.test) traj.f_add_link('hh', traj.test) traj.par.f_add_link('overview', traj.test) with self.assertRaises(ValueError): traj.f_add_link('overview', traj.test) with self.assertRaises(ValueError): traj.f_add_link('v_crun') with self.assertRaises(ValueError): traj.par.f_add_link('gg', traj) with self.assertRaises(AttributeError): traj.f_add_parameter('test.circle1.testy', 33) traj.par.f_add_link('gg', traj.circle1) self.assertTrue(traj.gg is traj.test2) self.assertTrue(traj.test2.test3 is traj.par.test.test3) traj.f_add_link(traj.test3) self.assertTrue('test3' in traj._links)
def test_storage_and_loading(self): filename = make_temp_dir('linktest.hdf5') traj = Trajectory(filename=filename) traj.f_add_parameter_group('test') traj.f_add_parameter_group('test2') res= traj.f_add_result('kk', 42) traj.par.f_add_link('gg', res) traj.f_add_link('hh', res) traj.f_add_link('jj', traj.par) traj.f_add_link('ii', res) traj.test.f_add_link('circle1' , traj.test2) traj.test2.f_add_link('circle2' , traj.test) traj.test.f_add_link('circle2' , traj.test.circle1.circle2) traj.f_add_parameter_group('test.ab.bc.cd') traj.cd.f_add_link(traj.test) traj.test.f_add_link(traj.cd) traj.f_store() traj2 = Trajectory(filename=filename) traj2.f_load(name=traj.v_name, load_data=2) self.assertTrue(traj.kk == traj2.gg, '%s != %s' % (traj.kk, traj2.gg)) self.assertTrue(traj.cd.test is traj.test) self.assertTrue(len(traj._linked_by), len(traj2._linked_by)) self.compare_trajectories(traj, traj2) self.assertTrue('jj' in traj2._nn_interface._links_count) traj2.f_remove_child('jj') self.assertTrue('jj' not in traj2._nn_interface._links_count) traj2.f_remove_child('hh') traj2.f_remove_child('ii') traj2.f_remove_child('parameters', recursive=True) traj2.v_auto_load = True group = traj2.par.test2.circle2 self.assertTrue(group is traj2.test) retest = traj2.test.circle1 self.assertTrue(retest is traj2.test2) self.assertTrue(traj2.test.circle2 is traj2.test) self.assertTrue(traj2.hh == traj2.res.kk) traj2.v_auto_load = False traj2.f_load_child('jj') self.assertTrue(traj2.jj is traj2.par) traj2.f_load(load_data=2) self.assertTrue(traj2.ii == traj2.res.kk)
def test_new_assignment_method(self): filename = make_temp_dir('newassignment.hdf5') traj = Trajectory(filename=filename, add_time=True) comment = 'A number' traj.par.x = Parameter('', 44, comment) self.assertTrue(traj.f_get('x').v_comment == comment) traj.x = 45 self.assertTrue(traj.par.f_get('x').f_get() == 45) self.assertTrue(isinstance(traj.f_get('x'), Parameter)) with self.assertRaises(AttributeError): traj.f = Parameter('lll', 444, 'lll') traj.f = Parameter('', 444, 'lll') self.assertTrue(traj.f_get('f').v_name == 'f') conf = traj.conf with self.assertRaises(AttributeError): conf = traj.conf.jjjj traj.f_set_properties(fast_access=True) traj.crun = Result('', k=43, m='JJJ') self.assertTrue(traj.run_A['k'] == 43) with self.assertRaises(AttributeError): traj.f_set_properties(j=7) with self.assertRaises(AttributeError): traj.f_set_properties(depth=7) traj.hui = Result('hui', ('444', 'kkkk',), 'l') self.assertTrue(traj.f_get('hui')[1] == 'l') traj.f_get('hui').f_set(('445', 'kkkk',)) self.assertTrue(traj.f_get('hui')[1] == 'l') self.assertTrue(traj.hui[0] == ('445', 'kkkk',)) traj.f_add_link('klkikju', traj.par) # for shizzle traj.meee = Result('meee.h', 43, hui = 3213, comment='du') self.assertTrue(traj.meee.h.h == 43) with self.assertRaises(TypeError): traj.par.jj = NNGroupNode('jj', comment='mi') with self.assertRaises(TypeError): traj.res.jj = NNGroupNode('jj', comment='mi') with self.assertRaises(TypeError): traj.conf.jj = NNGroupNode('jj', comment='mi') with self.assertRaises(TypeError): traj.dpar.jj = NNGroupNode('jj', comment='mi') with self.assertRaises(TypeError): traj.par.jj = ResultGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.dpar.jj = ResultGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.conf.jj = ResultGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.jj = ResultGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.par.jj = ConfigGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.dpar.jj = ConfigGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.res.jj = ConfigGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.jj = ConfigGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.par.jj = DerivedParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.conf.jj = DerivedParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.res.jj = DerivedParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.jj = DerivedParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.dpar.jj = ParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.res.jj = ParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.conf.jj = ParameterGroup('jj', comment='mi') with self.assertRaises(TypeError): traj.jj = ParameterGroup('jj', comment='mi') traj.par.jj = ParameterGroup('jj', comment='mi') traj.res.jj = ResultGroup('jj', comment='mi') traj.jj = NNGroupNode('jj') cg = ConfigGroup('a.g') traj.conf.a = cg self.assertTrue(traj.f_get('conf.a.g', shortcuts=False) is cg) dg = DerivedParameterGroup('ttt') traj.dpar.ttt = dg self.assertTrue(traj.f_get('dpar.ttt', shortcuts=False) is dg) traj.mylink = traj.par self.assertTrue(traj.mylink is traj.par) traj.vvv = NNGroupNode('', comment='kkk') self.assertTrue(traj.vvv.v_full_name == 'vvv') self.assertTrue(traj.par.jj.v_name == 'jj') traj.ff = MyParamGroup('ff') traj.par.g = MyParamGroup('') pg = traj.f_add_parameter_group(comment='gg', full_name='me') self.assertTrue(traj.par.me is pg) traj.f_store() traj = load_trajectory(index=-1, filename=filename, dynamic_imports=MyParamGroup) self.assertTrue(isinstance(traj.ff, MyParamGroup)) self.assertTrue(isinstance(traj.par.g, MyParamGroup)) traj.par.hiho = Parameter('hiho', 42, comment='you') traj.par.g1 = Parameter('g1.g2.g3.g4.g5', 43) self.assertTrue(traj.hiho == 42) self.assertTrue(isinstance(traj.par.g1, ParameterGroup )) self.assertTrue(isinstance(traj.par.g3, ParameterGroup )) self.assertTrue(traj.g3.g5 == 43)