def setUp(self): print("Setup...") UserSQL.create_table() self.test_user1 = User(id=1, userID='admin', password='******', admin=True) self.test_user2 = User(id=2, userID='user', password='******', admin=False) UserSQL.insert(self.test_user1) UserSQL.insert(self.test_user2)
import sys sys.path.append("..") from project.user.models import User from database import UserSQL UserSQL.create_table() test_user1 = User(id=1, userID='admin', password='******', admin=True) test_user2 = User(id=2, userID='user', password='******', admin=False) UserSQL.insert(test_user1) UserSQL.insert(test_user2)