def test_run_tessphot_mpi(PRIVATE_INPUT_DIR):
stars_to_test = ([260795451, 'tpf'], [260795451, 'ffi'])
with TaskManager(PRIVATE_INPUT_DIR) as tm:
tm.cursor.execute("UPDATE todolist SET status=1;")
for starid, datasource in stars_to_test:
tm.cursor.execute(
"UPDATE todolist SET status=NULL WHERE starid=? AND datasource=?;",
[starid, datasource])
tm.conn.commit()
tm.cursor.execute(
"SELECT COUNT(*) FROM todolist WHERE status IS NULL;")
num = tm.cursor.fetchone()[0]
print(num)
assert num == len(stars_to_test)
out, err, exitcode = capture_cli(
'run_tessphot_mpi.py',
mpiexec=True,
params=['--debug', '--version=0', PRIVATE_INPUT_DIR])
assert ' - INFO - %d tasks to be run' % num in err
assert ' - INFO - Master starting with 1 workers' in err
assert ' - DEBUG - Got data from worker 1: {' in err
assert ' - INFO - Worker 1 exited.' in err
assert ' - INFO - Master finishing' in err
assert exitcode == 0
def test_run_prepare_photometry(PRIVATE_INPUT_DIR):
hdf5file = os.path.join(PRIVATE_INPUT_DIR, 'sector001_camera3_ccd2.hdf5')
os.remove(hdf5file)
out, err, exitcode = capture_cli('run_prepare_photometry.py', params=['--camera=3', '--ccd=2', PRIVATE_INPUT_DIR])
assert exitcode == 0
assert os.path.isfile(hdf5file), "HDF5 was not created"
hdf5_file_valid(hdf5file)
def test_run_ffimovie(SHARED_INPUT_DIR):
out, err, exitcode = capture_cli('run_ffimovie.py',
params=[SHARED_INPUT_DIR])
assert exitcode == 0
for fname in ('sector001_camera3_ccd2.mp4',
'sector001_combined_backgrounds.mp4',
'sector001_combined_flags.mp4',
'sector001_combined_images.mp4',
'sector001_combined_originals.mp4'):
mp4file = os.path.join(SHARED_INPUT_DIR, fname)
assert os.path.isfile(mp4file), "MP4 was not created: " + fname
def test_make_todolist_cli(PRIVATE_INPUT_DIR):
# Path to the TODO-file:
todofile = os.path.join(PRIVATE_INPUT_DIR, 'todo.sqlite')
# Delete existing todo-file:
os.remove(todofile)
# Run make_todo CLI script:
out, err, exitcode = capture_cli(
'make_todo.py', params=['--camera=3', '--ccd=2', PRIVATE_INPUT_DIR])
assert exitcode == 0
assert '- ERROR -' not in out
assert '- ERROR -' not in err
assert '- INFO - TODO done.' in err
assert os.path.isfile(todofile), "TODO-file was not created"
todo_file_valid(todofile)
def test_run_tessphot_mpi_invalid_ccd():
out, err, exitcode = capture_cli('run_tessphot_mpi.py',
params=["--ccd=14"])
assert exitcode == 2
assert 'error: argument --ccd: invalid choice: 14 (choose from 1, 2, 3, 4)' in err
def test_run_tessphot_mpi_invalid_datasource():
out, err, exitcode = capture_cli('run_tessphot_mpi.py',
params=["--datasource=invalid"])
assert exitcode == 2
assert "error: argument --datasource: invalid choice: 'invalid'" in err