def test_get_pairs_dict(self):
pairs_list = [{
'11 сентября': [
Lecture.cell_parser(''),
Lecture.cell_parser(''),
Lecture.cell_parser(
'Безопасность ОС:\nПЗ-15 ауд. 122\nгр. 7334'),
Lecture.cell_parser(''),
]
}, {
'02 октября': [
Lecture.cell_parser(''),
Lecture.cell_parser(''),
Lecture.cell_parser(
'Безопасность ОС: \nлаб.раб-1 ауд. 122\nгр. 7334'),
Lecture.cell_parser(
'Безопасность ОС: \nсем.-1 ауд. 209\nгр. 7334'),
]
}, {
'06 ноября': [
Lecture.cell_parser(''),
Lecture.cell_parser(''),
Lecture.cell_parser(''),
Lecture.cell_parser(''),
]
}]
for (pairs, df) in zip(pairs_list, self.init_dataframes):
teacher = Teacher.df_parser(df)
self.assertEqual(pairs, teacher.get_pairs_dict())
def serialize(excel_file):
jsonobj = JSONSerializer()
jsonobj._xl_file = ExcelParser(excel_file)
jsonobj._names = jsonobj._xl_file.get_teachers_list()
jsonobj._teachers_pairs_dict = {}
for teacher_name in jsonobj._names:
df = jsonobj._xl_file.get_teacher_df(teacher_name)
teacher = Teacher.df_parser(df)
jsonobj._teachers_pairs_dict[
teacher_name] = teacher.get_pairs_dict()
jsonobj._dates += teacher.get_dates_list()
for date in set(jsonobj._dates):
jsonobj._date_dict[date] = {i: [] for i in range(1, 5)}
for teacher_name in jsonobj._names:
for date in jsonobj._teachers_pairs_dict[teacher_name]:
for i, pair in enumerate(
jsonobj._teachers_pairs_dict[teacher_name][date]):
pair_list = pair.to_list(teacher_name)
if pair_list:
jsonobj._date_dict[date][i + 1].append(pair_list)
return jsonobj
def test_get_dates_list(self):
lists = [['11 сентября'], ['02 октября'], ['06 ноября']]
for (l, df) in zip(lists, self.init_dataframes):
teacher = Teacher.df_parser(df)
self.assertEqual(l, teacher.get_dates_list())