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)