Modify assertEqual to use before_add / after_add