def test_import_nonexistent(INPUT_DIR):
"""
Tests that BaseCorrector handles being called with non-existing input directory.
"""
with pytest.raises(FileNotFoundError):
with BaseCorrector(INPUT_DIR + '/does/not/exist/'):
pass
def test_import():
"""
Tests if the module can even be imported.
Doesn't really do anything else..."""
with BaseCorrector(INPUT_DIR) as bc:
assert bc.__class__.__name__ == 'BaseCorrector', "Did not get the correct class name back"
assert bc.input_folder == INPUT_DIR, "Incorrect input folder"
def test_search_database_changes(PRIVATE_TODO_FILE):
"""
Test wheter changing corr_status will change what is returned by search_database.
"""
with BaseCorrector(PRIVATE_TODO_FILE) as bc:
rows1 = bc.search_database(search=['todolist.starid=29281992'])
print(rows1)
with TaskManager(PRIVATE_TODO_FILE) as tm:
task = tm.get_task(priority=17)
tm.start_task(task)
with BaseCorrector(PRIVATE_TODO_FILE) as bc:
rows2 = bc.search_database(search=['todolist.starid=29281992'])
print(rows2)
assert len(rows1) == len(rows2)
# Only the corr_status column was allowed to change!
assert rows1[0]['corr_status'] is None
assert rows2[0]['corr_status'] == STATUS.STARTED.value
for k in range(len(rows1)):
r1 = rows1[k]
r2 = rows2[k]
r1.pop('corr_status')
r2.pop('corr_status')
# For the test-data the "method_used" column should appear after
# the TaskManager has been run:
assert 'method_used' not in r1
assert r2['method_used'] == 'aperture'
r2.pop('method_used')
# For the test-data the cadence is added by TaskManager when initializing it:
assert 'cadence' not in r1
assert r2['cadence'] in (120, 1800)
r2.pop('cadence')
assert r1 == r2
def test_search_database(PRIVATE_TODO_FILE):
"""
Tests the search_database method of BaseCorrector.
.. codeauthor:: Rasmus Handberg <*****@*****.**>
"""
test_priority = 17
with BaseCorrector(PRIVATE_TODO_FILE) as bc:
res1 = bc.search_database(search='todolist.priority=%d' %
test_priority)
print(res1)
assert isinstance(res1, list), "Expects to get a list of results"
assert len(res1) == 1, "There should only be one result for this query"
assert isinstance(res1[0], dict), "Expects each result to be a dict"
res2 = bc.search_database(
search=['todolist.priority=%d' % test_priority], limit=10)
print(res2)
assert res1 == res2, "Should give same result with list"
res3 = bc.search_database(search=('todolist.priority=%d' %
test_priority, ),
limit=5)
print(res3)
assert res1 == res3, "Should give same result with tuple"
res4 = bc.search_database(search='todolist.priority=%d' %
test_priority,
distinct=True)
print(res4)
assert res1 == res4, "Should give same result with distinct=True"
# Make sure that we are not able to change anything in the TODO-file:
with pytest.raises(sqlite3.OperationalError):
bc.cursor.execute(
"UPDATE todolist SET priority=-{0:d} WHERE priority={0:d};".
format(test_priority))
bc.conn.rollback()
logger.addHandler(console)
logger.setLevel(logging_level)
logger_parent = logging.getLogger('corrections')
logger_parent.addHandler(console)
logger_parent.setLevel(logging_level)
# Parse the input folder:
input_folder = args.input_folder
if input_folder is None:
input_folder = os.environ.get('TESSCORR_INPUT')
if input_folder is None or not os.path.isdir(input_folder):
parser.error("Invalid input folder")
logger.info("Loading input data from '%s'", input_folder)
# Use the BaseCorrector to search the database for which CBV_AREAS to run:
with BaseCorrector(input_folder) as bc:
# Build list of constraints:
constraints = []
if args.camera:
constraints.append('camera IN (%s)' %
",".join([str(c) for c in args.camera]))
if args.ccd:
constraints.append('ccd IN (%s)' %
",".join([str(c) for c in args.ccd]))
if args.area:
constraints.append('cbv_area IN (%s)' %
",".join([str(c) for c in args.area]))
if not constraints:
constraints = None
# Search for valid areas: