def test_delete_is_working(self): with DB() as database: test = Test().create() test.id = database.execute('''SELECT id FROM tests WHERE title = ?''', \ (test.title,)).fetchone()[0] with DB() as database: test.delete() self.assertEqual(Test.find(test.id), None)