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