def test_save_pickle_try_big_data(self):
self.data = [a for a in range(30000)]
self.data += ["hello {}".format(a) for a in range(30000)]
db.renew_pickle(self.data, self.file_name)
self.assertTrue(os.path.exists(self.file_name))
def test_save_pickle_on_debug(self):
# self.data = ''
db.renew_pickle(self.data, self.file_name)
# print(''.join(self.out))
# print(self.out)
self.assertEqual("".join(self.out), "just for test\n")
def test_save_pickle(self):
# проверяем не было ли файла до этого
self.assertFalse(os.path.exists(self.file_name))
# проверяемый функционал
db.renew_pickle(self.data, self.file_name)
# проверяем - был ли создан файл
self.assertTrue(os.path.exists(self.file_name))
# проверка целостности
self.assertEqual(db.read_pickle_object(self.file_name), self.data)
def insert_auto_type_into_db(raw_brand):
global data
brand = 0
what = ''
auto = tuple()
#если бренд не был передан - запускаем блок заполнения бренда
if raw_brand == '':
raw_brand = take_from_input('введите марку : ').strip()
if raw_brand == 'выход':
return False
#если вводимый бренд(марка) состоит только из букв
if raw_brand.isalpha():
#если ещё не было подобного бренда - добавляем
if raw_brand not in data['raw_brands']:
data['raw_brands'].append(raw_brand)
#обновляем объект
data = renew_pickle(data, file_name)
insert_auto_type_into_db(raw_brand)
#если бренд подобный уже был - перезапускаем функцию с заполненным брендом
else:
insert_auto_type_into_db(raw_brand)
else :
#исключение - перезапуск функции
print_error_message(3)
insert_auto_type_into_db('')
else :
#brand есть - идём дальше
brand = data['raw_brands'].index(raw_brand)
#получаем мощность
power = take_from_input('введите мощность : ')
#если ввели не цифры - выходим - перезапускаем
if not power.isnumeric:
print_error_message(5)
insert_auto_type_into_db(raw_brand)
else:
#если новообразованный сет имеет разницу с текущим - добавляем его
if not data['auto_type'] or not data['auto_type'].intersection( { tuple([brand, int(power)], ) } ):
#if not data['auto_type'] or data['auto_type'].difference( { tuple([brand, int(power)], ) } ):
print('новый авто добавлен')
data['auto_type'].add( tuple( [brand, int(power)]) )
data = renew_pickle(data, file_name)
else:
print_error_message(6)
def delete_auto_type(id): data['auto_type'].remove(id) data = renew_pickle(data, file_name)
def test_save_pickle_try_empty(self):
self.data = ""
db.renew_pickle(self.data, self.file_name)
self.assertTrue(os.path.exists(self.file_name))
