class TestFunc(unittest.TestCase):
def setUp(self):
self.db = Database(provider='sqlite', filename=':memory:')
define_entity(self.db)
self.db.generate_mapping(create_tables=True)
create_entities(self.db, user_data)
def tearDown(self):
self.db.drop_all_tables(with_all_data=True)
@db_session
def test_create_entities(self):
# pylint: disable=E1101
user = self.db.User.get(username="******")
self.assertEqual(user.location, 'London')
metadata = Required(Json, default={})
created_at = Required(dt, default=lambda: dt.utcnow(), index=True)
updated_at = Required(dt, default=lambda: dt.utcnow())
performances = Set('Performance')
class Performance(db.Entity):
id = PrimaryKey(UUID, default=uuid4, auto=True)
ticks = Required(int)
crosses = Required(int)
unmarked = Required(int)
marks_obtained = Required(int)
total_marks = Required(int)
total_number_of_questions = Required(int)
percentage = Required(int)
metadata = Required(Json, default={})
created_at = Required(dt, default=lambda: dt.utcnow(), index=True)
updated_at = Required(dt, default=lambda: dt.utcnow())
grade = Required(Grade)
exam = Required(Exam)
user = Required(User)
composite_key(user, exam)
db.generate_mapping(create_tables=True)
if __name__ == '__main__':
db.generate_mapping(create_tables=True)
db.drop_all_tables(with_all_data=True)
