def test_init_6(self): from shortening.service.models import db, User, Shortening from peewee import IntegrityError # Given... db = SqliteDatabase(TEST_DB, check_same_thread=False) db.connect() # remove all records Shortening.delete().execute() # Test... total = 1000 t0 = time.time() for x in range(total): success = False while not success: try: s = Shortening(user=User.get(login="******"), short_url=Shortening.generate_random_shortening(), long_url="http://google.com/") s.save() except IntegrityError: pass else: success = True # Shortening already exists t1 = time.time() diff = t1 - t0 diff_per_row = diff / total print("Insert records with random name performance test, method 3, set size: %s, time: %s s, per row: %s s" % (total, diff, diff_per_row)) db.close()