def test_generate_raw_schedule_technician_multiple():
"""Tests function returns list for technician with multiple worksheets."""
app_config = {
't_excel': {
'sheet': ['test1', 'test2'],
'name_row': 0,
'col_start': 1,
'col_end': 100,
'row_start': 1,
'row_end': 100,
'date_col': 0,
'ext': 'xls',
}
}
excel_files = {'t': '/fake/path'}
user = {
'role': 't',
'schedule_name': 'test',
}
schedule = extract_schedule.generate_raw_schedule(app_config, excel_files,
user)
print(schedule)
assert len(schedule) == 2
def assemble_schedule(app_config, excel_files, user):
"""Assembles all the schedule details for provided user."""
old_schedule = retrieve_old_schedule(app_config, user['sb_user'])
new_schedule_raw = generate_raw_schedule(app_config, excel_files, user)
new_schedule = Schedule(old_schedule, new_schedule_raw, user, app_config)
new_schedule.process_new_schedule()
return new_schedule
def test_generate_raw_schedule_pharmacist_multiple():
"""Tests function returns list for pharmacist with multiple worksheets."""
app_config = {
'p_excel': {
'sheet': ['test1', 'test2'],
'name_row': 0,
'col_start': 1,
'col_end': 100,
'row_start': 1,
'row_end': 100,
'date_col': 0,
'ext': 'xlsx',
}
}
excel_files = {'p': '/fake/path'}
user = {
'role': 'p',
}
schedule = extract_schedule.generate_raw_schedule(app_config, excel_files,
user)
assert len(schedule) == 2