exclude_set = set(('hdf5_settings', 'multiproc', 'merge'))
integration_pred = lambda class_name, test_name, tags: (
'integration' in tags and not bool(exclude_set & tags))
integration_suite = discover_tests(integration_pred)
include_set = set(('hdf5_settings', 'links', 'merge'))
integration_pred_2 = lambda class_name, test_name, tags: (
'integration' in tags and bool(include_set & tags) and 'multiproc' not in
tags and 'links' not in tags)
integration_suite_2 = discover_tests(integration_pred_2)
suite_dict = {
'1': unit_suite,
'2': integration_suite,
'3': integration_suite_2
}
if __name__ == '__main__':
opt_dict = parse_args()
suite = None
if 'suite_no' in opt_dict:
suite_no = opt_dict.pop('suite_no')
suite = suite_dict[suite_no]
if suite is None:
pred = lambda class_name, test_name, tags: (
'multiproc' not in tags and class_name != TEST_IMPORT_ERROR)
suite = discover_tests(pred)
run_suite(suite=suite, **opt_dict)
filename = make_temp_dir('locker_test/scoop.txt')
self.create_file(filename)
self.start_server(url)
lock = LockerClient(url)
lock.start()
iterator = [(irun, lock, filename) for irun in range(self.ITERATIONS)]
list(futures.map(the_job, iterator))
lock.send_done()
self.check_file(filename)
self.lock_process.join()
# errwrite(str(irun))
self.ITERATIONS = old_iter
def test_timout_pool(self):
pool = mp.Pool(5)
url = get_random_port_url()
self.start_timeout_server(url, 0.25)
lock = LockerClient(url)
lock.start()
iterator = [(irun, lock) for irun in range(self.ITERATIONS)]
potential_timeouts = list(pool.imap(time_out_job, iterator))
pool.close()
pool.join()
all_time_outs = [x for x in potential_timeouts if x]
self.assertGreaterEqual(len(all_time_outs), 1)
lock.send_done()
self.lock_process.join()
if __name__ == '__main__':
opt_args = parse_args()
run_suite(**opt_args)
__author__ = 'robert'
try:
import pypet
except ImportError:
import sys
sys.path.append('/media/data/PYTHON_WORKSPACE/pypet-project')
import scoop
from pypet.tests.testutils.ioutils import discover_tests, parse_args, run_suite
from pypet.tests.integration.environment_scoop_test import check_mock
scoop_suite = discover_tests(lambda class_name, test_name, tags: 'scoop' in tags)
if __name__ == '__main__':
mock = check_mock()
if mock:
raise RuntimeError('Not running in SCOOP mode!')
opt_dict = parse_args()
run_suite(suite=scoop_suite, **opt_dict)
with self.assertRaises(ValueError):
node.v_annotations.f_get()
with self.assertRaises(AttributeError):
node.v_annotations.f_get('gdsdfd')
testparam = self.traj.f_add_parameter('ggg', 343)
with self.assertRaises(AttributeError):
testparam.v_annotations.f_get()
def test_f_set_numbering(self):
int_list = list(range(10))
for node in self.traj.f_iter_nodes(recursive=True):
node.v_annotations.f_set(*int_list)
self.assertEqual(node.v_annotations.f_get(*int_list),
tuple(int_list))
for integer in int_list:
if integer == 0:
name = 'annotation'
else:
name = 'annotation_%d' % integer
self.assertTrue(name in node.v_annotations)
if __name__ == '__main__':
opt_args = parse_args()
run_suite(**opt_args)
