Use assertTrue instead of assert.