from models import User
from app_factory import db


def add_to_database(object):
    db.session.add(object)
    db.session.commit()

db.session.close()
db.drop_all()
db.create_all()

add_to_database(User(email='test_email', password='******'))
add_to_database(User(email='*****@*****.**', password='******'))

 def tearDown(self):
     db.session.close()
     db.drop_all()
     db.create_all()
     assert len(User.query.all()) == 0
 def tearDown(self):
     db.session.close()
     db.drop_all()
     db.create_all()
     assert len(User.query.all()) == 0