def test_2(self):
try:
subprocess.Popen('ipcluster start --cluster-id=testclusterid --n=1 --daemonize', shell=True)
time.sleep(5)
args = self.Args()
args.output_dir = tempfile.mkdtemp()
args.cluster_id = 'testclusterid'
logger = AssignParallel.setup_logger()
AssignParallel.main(args, logger)
assignments = Serializer.LoadData(os.path.join(args.output_dir, 'Assignments.h5'))
r_assignments = Serializer.LoadData(os.path.join(fixtures_dir(), 'Assignments.h5'))
distances = Serializer.LoadData(os.path.join(args.output_dir, 'Assignments.h5.distances'))
r_distances = Serializer.LoadData(os.path.join(fixtures_dir(), 'Assignments.h5.distances'))
npt.assert_array_equal(assignments, r_assignments)
npt.assert_array_almost_equal(distances, r_distances)
except:
raise
finally:
shutil.rmtree(args.output_dir)
subprocess.Popen('ipcluster stop', shell=True).wait()
def test_setup_logger():
progname = 'TEST PROGNAME'
message = 'TEST MESSAGE'
sys.argv[0] = progname
fid, path = tempfile.mkstemp()
stream = open(path, 'w')
logger = AssignParallel.setup_logger(console_stream=stream)
logger.info(message)
stream.flush()
line = open(path).readlines()[0]
os.unlink(path)
ok_(line.startswith(progname))
ok_(line.endswith(message + '\n'))
